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

  /external/tensorflow/tensorflow/cc/gradients/
data_flow_grad.cc 72 auto partitions_shape = Shape(scope, partitions); local
82 scope, Range(scope, zero, Prod(scope, partitions_shape, zero), one),
83 partitions_shape);
  /external/tensorflow/tensorflow/core/ops/
data_flow_ops.cc 59 ShapeHandle partitions_shape = c->input(1);
61 if (!c->RankKnown(partitions_shape)) {
65 const int64 rank = c->Rank(partitions_shape);
67 // data shape must start with partitions_shape
70 c->MergePrefix(data_shape, partitions_shape, &unused, &unused));
    [all...]

Completed in 83 milliseconds