| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_meta_ops.cc | 86 shape_inference::DimensionHandle out_dim; in __anon82f11ae70102() local
88 /*evenly_divisible=*/true, &out_dim)); in __anon82f11ae70102()
89 out_dims.push_back(out_dim); in __anon82f11ae70102()
143 shape_inference::DimensionHandle out_dim; in __anon82f11ae70202() local
144 TF_RETURN_IF_ERROR(c->Multiply(dim, input_sharding[i], &out_dim)); in __anon82f11ae70202()
145 out_dims.push_back(out_dim); in __anon82f11ae70202()
|
| /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 | 1350 for (int64_t out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local
1351 const int64_t in_dim = out_dim + offset; in ComputeFlatInnerDims()
1352 out_dims[out_dim] = in_dim < 0 ? 1 : orig[in_dim]; in ComputeFlatInnerDims()
1363 for (int64_t out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local
1364 out_dims[out_dim] = out_dim >= orig.size() ? 1 : orig[out_dim]; in ComputeFlatOuterDims()
|
| D | common_shape_fns.cc | 2626 shape_inference::DimensionHandle out_dim; in ReduceScatterShape() local
2628 /*evenly_divisible=*/true, &out_dim)); in ReduceScatterShape()
2629 out_dims.push_back(out_dim); in ReduceScatterShape()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | DeconvolutionLayerFixture.h | 257 …auto out_dim = deconvolution_output_dimensions(input_shape.x(), input_shape.y(), ke… in setup() local
260 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup()
278 …auto out_dim = deconvolution_output_dimensions(input_shape.x(), input_shape.y(), ke… in setup() local
281 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup()
299 …auto out_dim = deconvolution_output_dimensions(input_shape.x(), input_shape.y(), ke… in setup() local
302 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup()
322 …auto out_dim = deconvolution_output_dimensions(input_shape.x(), input_shape.y(), ke… in setup() local
325 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op.cc | 176 int out_dim = dim + indices_in.dims() - 1; in MakeSplits() local
177 if (out_dim >= 0) { in MakeSplits()
178 SPLITS_TYPE delta = out_splits->at(out_dim).back() - splits(start); in MakeSplits()
180 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/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 196 for (int out_dim = 0; out_dim < NumDims; ++out_dim) { in test_block_io_copy_using_reordered_dimensions() local
197 dst_dims[output_to_input_dim_map[out_dim]] = blk_dims[out_dim]; in test_block_io_copy_using_reordered_dimensions()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops.cc | 442 const int64_t out_dim = output_slice_shape_dsizes_[dim]; in SliceAndMaybePadState() local
448 slice_paddings_[dim] = {0, out_dim}; in SliceAndMaybePadState()
450 } else if (slice_indices_[dim] + out_dim > dim_size) { in SliceAndMaybePadState()
453 slice_paddings_[dim] = {0, out_dim - non_padded_dim}; in SliceAndMaybePadState()
456 non_padded_dim = out_dim; in SliceAndMaybePadState()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_normalization.cc | 421 int out_dim = tr_dim - DegenDimsUpTo(normalized_w_degen, tr_dim); in HandlePad() local
422 *new_padding.mutable_dimensions(out_dim) = padded_config.dimensions(dim); in HandlePad()
|
| D | reduce_scatter_utils.cc | 498 absl::c_all_of(split_dims, [&](int64_t out_dim) { in MatchReduceScatter() argument
499 return unmodified_output_to_input_map.count(out_dim) != 0; in MatchReduceScatter()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_v1.py | 826 for target_dim, out_dim in zip(y.shape[1:], shape[1:]):
827 if out_dim is not None and target_dim != out_dim:
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 501 auto out_dim = c->Dim(out, out_rank - i - 1); in __anon091a71b70702() local
502 TF_RETURN_IF_ERROR(c->Merge(in_dim, out_dim, &out_dim)); in __anon091a71b70702()
504 c->ReplaceDim(out, out_rank - i - 1, out_dim, &out)); in __anon091a71b70702()
|