Searched refs:to_apply_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 160 const ProgramShape& to_apply_shape);
292 const Shape& updates_shape, const ProgramShape& to_apply_shape,
|
| D | shape_inference.cc | 2114 const Window& window, const ProgramShape& to_apply_shape) { in InferReduceWindowShape() argument
2116 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferReduceWindowShape()
3165 const Shape& updates_shape, const ProgramShape& to_apply_shape, in InferScatterShape() argument
3193 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferScatterShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1875 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in Scatter()
1880 to_apply_shape, dimension_numbers)); in Scatter()
2028 TF_ASSIGN_OR_RETURN(const ProgramShape& to_apply_shape, in ReduceWindowWithGeneralPadding()
2036 instr.window(), to_apply_shape)); in ReduceWindowWithGeneralPadding()
|