Searched refs:DynamicPartition (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | data_flow_grad_test.cc | 29 using ops::DynamicPartition;
53 auto y = DynamicPartition(scope_, data, partitions, 3); in TEST_F()
|
| D | data_flow_grad.cc | 90 DynamicPartition(scope, original_indices, partitions, num_partitions); in DynamicPartitionGrad()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DynamicPartition.pbtxt | 2 graph_op_name: "DynamicPartition"
52 <img style="width:100%" src="https://www.tensorflow.org/images/DynamicPartition.png" alt>
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_DynamicPartition.pbtxt | 2 graph_op_name: "DynamicPartition"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_test.cc | 173 static Graph* DynamicPartition(int num_partitions, int dim) { in DynamicPartition() function
197 test::Benchmark(#DEVICE, DynamicPartition<T>(num, dim)).Run(iters); \
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | data_flow_ops_test.cc | 92 TEST(DataFlowOpsTest, DynamicPartition) { in TEST() argument
|
| D | ops.pbtxt | 10060 name: "DynamicPartition"
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 403 HANDLE_OPERATORTYPENAME_CASE(DynamicPartition) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 1133 ops::DynamicPartition part(scope.WithOpName("partition"), input, indices, in TEST_F()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 936 name: "DynamicPartition"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 936 name: "DynamicPartition"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 7357 func DynamicPartition(scope *Scope, data tf.Output, partitions tf.Output, num_partitions int64) (ou… func
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 10834 name: "DynamicPartition"
|
| D | ops_history.v1.pbtxt | 22050 name: "DynamicPartition"
|
| D | ops_history.v2.pbtxt | 21679 name: "DynamicPartition"
|