Searched defs:out_dim (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_meta_ops.cc | 86 shape_inference::DimensionHandle out_dim; in __anon82f11ae70102() local
143 shape_inference::DimensionHandle out_dim; in __anon82f11ae70202() local
|
| /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
278 …auto out_dim = deconvolution_output_dimensions(input_shape.x(), input_shape.y(), ke… in setup() local
299 …auto out_dim = deconvolution_output_dimensions(input_shape.x(), input_shape.y(), ke… in setup() local
322 …auto out_dim = deconvolution_output_dimensions(input_shape.x(), input_shape.y(), ke… in setup() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 155 DimensionHandle out_dim = c.Dim(out, j); in InferShapes() local
|
| D | tensor.cc | 1350 for (int64_t out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local
1363 for (int64_t out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local
|
| D | common_shape_fns.cc | 2626 shape_inference::DimensionHandle out_dim; in ReduceScatterShape() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op.cc | 176 int out_dim = dim + indices_in.dims() - 1; in MakeSplits() local
|
| D | eigen_spatial_convolutions_test.cc | 1307 DSizes<ptrdiff_t, 3> out_dim; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reduce_scatter_utils.cc | 498 absl::c_all_of(split_dims, [&](int64_t out_dim) { in MatchReduceScatter()
|
| D | layout_normalization.cc | 421 int out_dim = tr_dim - DegenDimsUpTo(normalized_w_degen, tr_dim); in HandlePad() local
|
| /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
|
| /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
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 501 auto out_dim = c->Dim(out, out_rank - i - 1); in __anon091a71b70702() local
|