Home
last modified time | relevance | path

Searched defs:out_dim (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference_testutil.cc157 DimensionHandle out_dim = c.Dim(out, j); in InferShapes() local
Dtensor.cc1210 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local
1223 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local
/external/tensorflow/tensorflow/python/framework/
Dcpp_shape_inference.cc39 auto* out_dim = out->add_dim(); in ProtoFromShapeHandle() local
/external/tensorflow/tensorflow/core/kernels/
Dragged_gather_op.cc166 int out_dim = dim + indices_in.dims() - 1; in MakeSplits() local
Deigen_spatial_convolutions_test.cc1307 DSizes<ptrdiff_t, 3> out_dim; in TEST() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc494 auto out_dim = c->Dim(out, out_rank - i - 1); in __anon7c94107b0702() local