Home
last modified time | relevance | path

Searched defs:init_value_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc2294 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape()
2344 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape()
2354 const Shape& init_value_shape, const ProgramShape& scatter_shape) { in InferSelectAndScatterShape()
H A Dshape_inference_test.cc329 Shape init_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local