Searched refs:partitions_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | data_flow_grad.cc | 72 auto partitions_shape = Shape(scope, partitions); in DynamicPartitionGrad() local
82 scope, Range(scope, zero, Prod(scope, partitions_shape, zero), one), in DynamicPartitionGrad()
83 partitions_shape); in DynamicPartitionGrad()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | data_flow_ops.cc | 59 ShapeHandle partitions_shape = c->input(1); in __anon4a7ac08d0202() local
61 if (!c->RankKnown(partitions_shape)) { in __anon4a7ac08d0202()
65 const int64 rank = c->Rank(partitions_shape); in __anon4a7ac08d0202()
70 c->MergePrefix(data_shape, partitions_shape, &unused, &unused)); in __anon4a7ac08d0202()
|