Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreshape_decomposer.cc36 Shape new_input_shape = *output_aligned_input_shape; in HandleReshape() local
H A Dconvolution_4d_expander.cc102 Shape new_input_shape = input->shape(); in ExpandInstruction() local
H A Dalgebraic_simplifier.cc6637 Shape new_input_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dconv_canonicalization.cc75 Shape new_input_shape = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_pad_for_convolutions.cc207 Shape* new_input_shape; in TryResolvePaddedShapesForTensorCore() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DOnednnUtils.h65 auto new_input_shape = std::get<InputShape>(new_key); in hit_dynamic() local