Searched refs:select_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2125 const Shape& operand_shape, const ProgramShape& select_shape, in InferSelectAndScatterShape() argument
2132 if (select_shape.parameters_size() != 2) { in InferSelectAndScatterShape()
2136 select_shape.parameters_size()); in InferSelectAndScatterShape()
2138 const Shape& select_result_shape = select_shape.result(); in InferSelectAndScatterShape()
2146 select_shape.parameters(0))) { in InferSelectAndScatterShape()
2150 ShapeUtil::HumanString(select_shape.parameters(0)), in InferSelectAndScatterShape()
2154 select_shape.parameters(1))) { in InferSelectAndScatterShape()
2158 ShapeUtil::HumanString(select_shape.parameters(1)), in InferSelectAndScatterShape()
|
| D | hlo_creation_utils.cc | 281 TF_ASSIGN_OR_RETURN(Shape select_shape, in MakeSelectHlo()
285 select_shape, HloOpcode::kSelect, pred, on_true, on_false)); in MakeSelectHlo()
|
| D | shape_inference.h | 165 const Shape& operand_shape, const ProgramShape& select_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2273 TF_ASSIGN_OR_RETURN(const ProgramShape& select_shape, in SelectAndScatterWithGeneralPadding()
2282 operand_shape, select_shape, instr.window(), in SelectAndScatterWithGeneralPadding()
|