Searched refs:out_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 155 DimensionHandle out_dim = c.Dim(out, j); in InferShapes() local
161 if (c.Dim(in, j).SameHandle(out_dim)) { in InferShapes()
175 } else if (c.ValueKnown(out_dim)) { in InferShapes()
177 c.Value(out_dim), err_suffix); in InferShapes()
218 } else if (value != c.Value(out_dim)) { in InferShapes()
220 c.DebugString(out_dim), err_suffix); in InferShapes()
|
| D | tensor.cc | 1274 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local
1275 const int64 in_dim = out_dim + offset; in ComputeFlatInnerDims()
1276 out_dims[out_dim] = in_dim < 0 ? 1 : orig[in_dim]; in ComputeFlatInnerDims()
1287 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local
1288 out_dims[out_dim] = out_dim >= orig.size() ? 1 : orig[out_dim]; in ComputeFlatOuterDims()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op.cc | 170 int out_dim = dim + indices_in.dims() - 1; in MakeSplits() local
171 if (out_dim >= 0) { in MakeSplits()
172 SPLITS_TYPE delta = out_splits->at(out_dim).back() - splits(start); in MakeSplits()
174 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/keras/engine/ |
| D | training_utils_v1.py | 832 for target_dim, out_dim in zip(y.shape[1:], shape[1:]):
833 if out_dim is not None and target_dim != out_dim:
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 500 auto out_dim = c->Dim(out, out_rank - i - 1); in __anon42d741190702() local
501 TF_RETURN_IF_ERROR(c->Merge(in_dim, out_dim, &out_dim)); in __anon42d741190702()
503 c->ReplaceDim(out, out_rank - i - 1, out_dim, &out)); in __anon42d741190702()
|