OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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