Searched refs:to_apply_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 199 const ProgramShape& to_apply_shape);
206 const ProgramShape& to_apply_shape);
354 const ProgramShape& to_apply_shape,
|
| D | shape_inference.cc | 2295 const Window& window, const ProgramShape& to_apply_shape) { in InferReduceWindowShape() argument
2296 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferReduceWindowShape()
2305 const ProgramShape& to_apply_shape) { in InferReduceWindowShape() argument
2323 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, init_values, in InferReduceWindowShape()
2335 if (ShapeUtil::IsScalar(to_apply_shape.result())) { in InferReduceWindowShape()
3659 const ProgramShape& to_apply_shape, in InferScatterShape() argument
3779 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, init_element_shape_ptrs, in InferScatterShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2463 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in Scatter()
2467 operand_shapes, to_apply_shape, dimension_numbers)); in Scatter()
2721 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in ReduceWindowWithGeneralPadding()
2731 to_apply_shape)); in ReduceWindowWithGeneralPadding()
2765 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in ReduceWindowInternal()
2775 absl::MakeSpan(init_shapes), window, to_apply_shape)); in ReduceWindowInternal()
|