Home
last modified time | relevance | path

Searched defs:new_output_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc111 Shape new_output_shape = instruction->shape(); in ExpandInstruction() local
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc409 SmallVector<int64_t, 4> new_output_shape; in matchAndRewrite() local
568 SmallVector<int64_t, 4> new_output_shape; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_vectorize_convolutions.cc415 Shape new_output_shape = SplitShapeAtDim( in TryVectorizeConv() local
Dcudnn_pad_for_convolutions.cc199 Shape* new_output_shape; in TryResolvePaddedShapesForTensorCore() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc463 auto new_output_shape = in ValidateAndGetTiledExecuteOutputShape() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2512 TensorShapeProto* new_output_shape = in SimplifyCase() local