Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreshape_mover_test.cc422 auto pred_shape = ShapeUtil::MakeShape(PRED, {8, 7}); in TEST_F() local
456 auto pred_shape = ShapeUtil::MakeShape(PRED, {}); in TEST_F() local
H A Ddynamic_padder.cc296 const Shape pred_shape = in PadWithScalar() local
H A Dalgebraic_simplifier.cc2824 auto pred_shape = ShapeUtil::ChangeElementType(gather->shape(), PRED); in HandleGather() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dvalue_inference_test.cc518 auto pred_shape = ShapeUtil::MakeShape(PRED, {}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc4854 Shape pred_shape = ShapeUtil::ChangeElementType(hlo->shape(), PRED); in PreprocessHlos() local