Searched refs:out_dim (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 157 DimensionHandle out_dim = c.Dim(out, j); in InferShapes() local
163 if (c.Dim(in, j).SameHandle(out_dim)) { in InferShapes()
177 } else if (c.ValueKnown(out_dim)) { in InferShapes()
179 c.Value(out_dim), err_suffix); in InferShapes()
220 } else if (value != c.Value(out_dim)) { in InferShapes()
222 c.DebugString(out_dim), err_suffix); in InferShapes()
|
| D | tensor.cc | 1210 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local
1211 const int64 in_dim = out_dim + offset; in ComputeFlatInnerDims()
1212 out_dims[out_dim] = in_dim < 0 ? 1 : orig[in_dim]; in ComputeFlatInnerDims()
1223 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local
1224 out_dims[out_dim] = out_dim >= orig.size() ? 1 : orig[out_dim]; in ComputeFlatOuterDims()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op.cc | 166 int out_dim = dim + indices_in.dims() - 1; in MakeSplits() local
167 if (out_dim >= 0) { in MakeSplits()
168 int64 delta = out_splits->at(out_dim).back() - splits(start); in MakeSplits()
170 out_splits->at(out_dim).push_back(splits(j + 1) + delta); in MakeSplits()
|
| D | eigen_spatial_convolutions_test.cc | 1307 DSizes<ptrdiff_t, 3> out_dim; in TEST() local
1308 out_dim[0] = 1; in TEST()
1309 out_dim[1] = 4; in TEST()
1310 out_dim[2] = 12; in TEST()
1348 .reshape(out_dim), in TEST()
1359 .reshape(out_dim), in TEST()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 39 auto* out_dim = out->add_dim(); in ProtoFromShapeHandle() local
41 out_dim->set_size(c->Value(d)); in ProtoFromShapeHandle()
43 out_dim->set_size(-1); in ProtoFromShapeHandle()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils.py | 502 for target_dim, out_dim in zip(y.shape[1:], shape[1:]):
503 if out_dim is not None and target_dim != out_dim:
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 494 auto out_dim = c->Dim(out, out_rank - i - 1); in __anon7c94107b0702() local
495 TF_RETURN_IF_ERROR(c->Merge(in_dim, out_dim, &out_dim)); in __anon7c94107b0702()
497 c->ReplaceDim(out, out_rank - i - 1, out_dim, &out)); in __anon7c94107b0702()
|