Home
last modified time | relevance | path

Searched refs:expanded_dims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h226 Eigen::DSizes<IndexType, NDIMS> expanded_dims;
231 expanded_dims[0] = merged_dims[2]; // [O]
232 expanded_dims[1] = merged_dims[1]; // [I]
234 expanded_dims[2 + i] = spatial_dims[i];
240 expanded_dims[0] = merged_dims[2]; // [O]
241 expanded_dims[NDIMS - 1] = merged_dims[1]; // [I]
243 expanded_dims[1 + i] = spatial_dims[i];
252 in.reshape(merged_dims).shuffle(shuffling_perm).reshape(expanded_dims);
265 Eigen::DSizes<IndexType, 4> expanded_dims;
276 expanded_dims[0] = in.dimension(shuffle[0]);
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc475 HloInstruction* operand, absl::Span<const int64> expanded_dims) { in ExpandFirstDimIntoNDims() argument
477 CHECK_EQ(operand->shape().dimensions(0), Product(expanded_dims)); in ExpandFirstDimIntoNDims()
480 expanded_shape_dim_bounds.reserve(expanded_dims.size() + in ExpandFirstDimIntoNDims()
482 absl::c_copy(expanded_dims, std::back_inserter(expanded_shape_dim_bounds)); in ExpandFirstDimIntoNDims()
Dhlo_creation_utils.h252 HloInstruction* operand, absl::Span<const int64> expanded_dims);
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4776 nvinfer1::Dims expanded_dims; in ConvertPack() local
4777 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(tensor_dims, &expanded_dims)); in ConvertPack()
4791 params->converter, input, expanded_dims, params->validation_only, in ConvertPack()