Home
last modified time | relevance | path

Searched refs:new_output_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc111 Shape new_output_shape = instruction->shape(); in ExpandInstruction() local
113 new_output_shape.DeleteDimension(dim); in ExpandInstruction()
173 new_output_shape, {reshaped_input, reshaped_kernel})); in ExpandInstruction()
Dreshape_decomposer.cc47 Shape new_output_shape = *input_aligned_output_shape; in HandleReshape() local
48 TF_ASSIGN_OR_RETURN(auto r, MakeReshapeHlo(new_output_shape, operand)); in HandleReshape()
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc413 SmallVector<int64_t, 4> new_output_shape; in matchAndRewrite() local
415 new_output_shape.push_back(1); in matchAndRewrite()
418 new_output_shape.push_back(size); in matchAndRewrite()
422 RankedTensorType::get(new_output_shape, output_type.getElementType()); in matchAndRewrite()
572 SmallVector<int64_t, 4> new_output_shape; in matchAndRewrite() local
574 new_output_shape.push_back(1); in matchAndRewrite()
577 new_output_shape.push_back(size); in matchAndRewrite()
581 RankedTensorType::get(new_output_shape, output_type.getElementType()); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc209 Shape* new_output_shape; in TryResolvePaddedShapesForTensorCore() local
210 std::tie(new_input_shape, new_filter_shape, new_output_shape) = [&] { in TryResolvePaddedShapesForTensorCore()
231 new_output_shape->dimensions(dnums.output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
242 pad_dim(new_output_shape, dnums.output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
Dcudnn_vectorize_convolutions.cc464 Shape new_output_shape = SplitShapeAtDim( in TryVectorizeConv() local
469 {new_output_shape, ShapeUtil::MakeShape(U8, {0})}), in TryVectorizeConv()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc474 auto new_output_shape = in ValidateAndGetTiledExecuteOutputShape() local
499 new_output_shape[dimension_index] = in ValidateAndGetTiledExecuteOutputShape()
504 new_output_shape, cluster_func_output_type.getElementType()); in ValidateAndGetTiledExecuteOutputShape()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc232 const Shape new_output_shape = in PartitionGatherIndexPassthroughPartition() local
251 new_output_shape, output_grouped.sharding, batch_dims, in PartitionGatherIndexPassthroughPartition()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2546 TensorShapeProto* new_output_shape = in SimplifyCase() local
2550 *new_output_shape = in SimplifyCase()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2903 new_output_shape = array_ops.concat(
2905 output = array_ops.reshape(output, new_output_shape)