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
2188 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape()2235 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape()2246 const Shape& init_value_shape, const ProgramShape& scatter_shape) { in InferSelectAndScatterShape()3566 const Shape init_value_shape = in InferScatterShape() local
327 Shape init_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local