Searched refs:to_apply_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 168 const ProgramShape& to_apply_shape);
175 const ProgramShape& to_apply_shape);
323 const Shape& updates_shape, const ProgramShape& to_apply_shape,
|
| D | shape_inference.cc | 2189 const Window& window, const ProgramShape& to_apply_shape) { in InferReduceWindowShape() argument
2190 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferReduceWindowShape()
2199 const ProgramShape& to_apply_shape) { in InferReduceWindowShape() argument
2216 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, init_values, in InferReduceWindowShape()
2226 if (ShapeUtil::IsScalar(to_apply_shape.result())) { in InferReduceWindowShape()
3540 const Shape& updates_shape, const ProgramShape& to_apply_shape, in InferScatterShape() argument
3568 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferScatterShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2311 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in Scatter()
2316 to_apply_shape, dimension_numbers)); in Scatter()
2567 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in ReduceWindowWithGeneralPadding()
2577 to_apply_shape)); in ReduceWindowWithGeneralPadding()
2611 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in ReduceWindowInternal()
2621 absl::MakeSpan(init_shapes), window, to_apply_shape)); in ReduceWindowInternal()
|