Searched refs:expanded_dims (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_2d.h | 226 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/ |
D | hlo_creation_utils.cc | 475 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()
|
D | hlo_creation_utils.h | 252 HloInstruction* operand, absl::Span<const int64> expanded_dims);
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.cc | 4776 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()
|