Home
last modified time | relevance | path

Searched defs:new_output_shape (Results 1 – 4 of 4) 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/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc207 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.cc2496 TensorShapeProto* new_output_shape = in SimplifyCase() local