Home
last modified time | relevance | path

Searched defs:to_shape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbitcast_dtypes_expander.cc42 const Shape& to_shape = instruction->shape(); in ExpandInstruction() local
H A Dhlo_element_type_converter.cc88 const Shape& to_shape) { in ConvertTupleElements()
H A Dalgebraic_simplifier.h50 bool ReshapeIsBitcast(const Shape& from_shape, const Shape& to_shape) const { in ReshapeIsBitcast()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_symbolic_shape_optimization.cc185 ShapeComponentAnalysis& analysis, Value from, Value to_shape) { in isNonExpandingBroadcast()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.cc1278 const Shape& from_shape, const Shape& to_shape, in ReshapeLeavesDimensionsUnmodified()