HomeSort by relevance Sort by last modified time
    Searched refs:DynamicPartition (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/cc/gradients/
data_flow_grad_test.cc 29 using ops::DynamicPartition;
53 auto y = DynamicPartition(scope_, data, partitions, 3);
data_flow_grad.cc 47 // DynamicPartition only moves input values into various positions
90 DynamicPartition(scope, original_indices, partitions, num_partitions);
107 REGISTER_GRADIENT_OP("DynamicPartition", DynamicPartitionGrad);
  /external/tensorflow/tensorflow/core/kernels/
dynamic_partition_op_test.cc 42 TF_ASSERT_OK(NodeDefBuilder("myop", "DynamicPartition")
163 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "DynamicPartition")
172 static Graph* DynamicPartition(int num_partitions, int dim) {
196 test::Benchmark(#DEVICE, DynamicPartition<T>(num, dim)).Run(iters); \
  /external/tensorflow/tensorflow/core/ops/
data_flow_ops_test.cc 92 TEST(DataFlowOpsTest, DynamicPartition) {
93 ShapeInferenceTestOp op("DynamicPartition");
94 TF_ASSERT_OK(NodeDefBuilder("test", "DynamicPartition")
106 TF_ASSERT_OK(NodeDefBuilder("test", "DynamicPartition")
  /external/tensorflow/tensorflow/core/grappler/optimizers/
constant_folding_test.cc 725 // Add a DynamicPartition node to the graph
729 ops::DynamicPartition part(scope.WithOpName("partition"), input, indices,
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go 318 // <img style="width:100%" src="https://www.tensorflow.org/images/DynamicPartition.png" alt>
325 func DynamicPartition(scope *Scope, data tf.Output, partitions tf.Output, num_partitions int64) (outputs []tf.Output) {
331 Type: "DynamicPartition",
344 scope.UpdateErr("DynamicPartition", err)
    [all...]

Completed in 732 milliseconds