Searched defs:init_value_shape (Results 1 – 3 of 3) sorted by relevance
57 const TensorShape init_value_shape = context->InputShape(n_); in Compile() local
2302 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape()2349 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape()2360 const Shape& init_value_shape, const ProgramShape& scatter_shape) { in InferSelectAndScatterShape()3703 const Shape init_value_shape = in InferScatterShape() local
327 Shape init_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local