Searched refs:select_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2244 const Shape& operand_shape, const ProgramShape& select_shape, in InferSelectAndScatterShape() argument
2251 if (select_shape.parameters_size() != 2) { in InferSelectAndScatterShape()
2255 select_shape.parameters_size()); in InferSelectAndScatterShape()
2257 const Shape& select_result_shape = select_shape.result(); in InferSelectAndScatterShape()
2265 select_shape.parameters(0))) { in InferSelectAndScatterShape()
2269 ShapeUtil::HumanString(select_shape.parameters(0)), in InferSelectAndScatterShape()
2273 select_shape.parameters(1))) { in InferSelectAndScatterShape()
2277 ShapeUtil::HumanString(select_shape.parameters(1)), in InferSelectAndScatterShape()
|
| D | shape_inference.h | 184 const Shape& operand_shape, const ProgramShape& select_shape,
|
| D | hlo_creation_utils.cc | 404 TF_ASSIGN_OR_RETURN(Shape select_shape, in MakeSelectHlo()
409 select_shape, select_op_code, pred, on_true, on_false)); in MakeSelectHlo()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 3082 TF_ASSIGN_OR_RETURN(const ProgramShape& select_shape, in SelectAndScatterInternal()
3092 *operand_shape, select_shape, instr.window(), in SelectAndScatterInternal()
|