Searched defs:init_value_shape (Results 1 – 3 of 3) sorted by relevance
45 const TensorShape init_value_shape = context->InputShape("init_value"); in Compile() local
2113 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape()2127 const Shape& init_value_shape, const ProgramShape& scatter_shape) { in InferSelectAndScatterShape()3191 const Shape init_value_shape = in InferScatterShape() local
315 Shape init_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local