experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
mm-fix-macos-protobuf-segfault
STT-tensorflow/tensorflow/core/api_def
History
Ayush Dubey e38610399a Add CollectiveBcastSend/RecvV2 which takes input tensors rather than attributes. 
CollectiveBcastSend/Recv accepts the following inputs as attributes on the op:
group_size, group_key, and instance_key.  Attributes imply these values are
embedded in the NodeDef.

The use case motivating this change is a compact representation for SPMD
computation.  The goal is to change those inputs from the collective op which
can be accepted during runtime to tensors rather than attributes.  This enables
the graph builder to avoid early explosion of the SPMD program.

This op is not exposed in the `tf.` namespace for now, and should be considered
experimental.

PiperOrigin-RevId: 348049802
Change-Id: I15ffcb562fb75cc64e5970ac7795cb81ff187fd8
2020-12-17 10:45:19 -08:00
..
base_api Add CollectiveBcastSend/RecvV2 which takes input tensors rather than attributes. 2020-12-17 10:45:19 -08:00
java_api Create a V2 Op to stop the gradient when the input is out of range. 2020-10-12 10:39:36 -07:00
python_api Rolling forward change by fixing issue with lack of dispatch for & and | 2020-12-02 00:09:03 -08:00
api_test.cc Make use of GetDataDependencyFilepath and io::JoinPath to be operating system independent. 2020-02-13 15:39:58 -08:00
BUILD Create BUILD files and corresponding targets for tensorflow/core/api_def/base_api. 2020-11-20 16:06:49 -08:00
excluded_ops.cc
excluded_ops.h Script to create ApiDef files automatically based on OpDef. 2018-01-18 17:19:42 -08:00
README.md Provide information to users about where the canonical documentation of ops is. 2020-09-21 09:09:48 -07:00
update_api_def_main.cc Script to create ApiDef files automatically based on OpDef. 2018-01-18 17:19:42 -08:00
update_api_def_test.cc Fix update_api_def.sh script to output ApiDefs instead of ApiDef proto. 2018-01-24 10:59:18 -08:00
update_api_def.cc Use absl instead of deprecated str_util 2019-05-21 08:10:10 -07:00
update_api_def.h Remove THIRD_PARTY_ from #include guards 2018-01-24 14:31:28 -08:00
update_api_def.sh Script to create ApiDef files automatically based on OpDef. 2018-01-18 17:19:42 -08:00

README.md

This folder contains the ApiDef proto definitions of TensorFlow operations.

The canonical source of documentation for these operations can be found in the base_api/ directory.