Searched refs:scatter_dimension_numbers (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 223 scatter->scatter_dimension_numbers(); in ScatterLoopBody()
351 scatter->scatter_dimension_numbers(); in ExpandScatter()
|
| D | hlo_instructions.cc | 2411 StrJoin(scatter_dimension_numbers().update_window_dims(), ","), "}"); in ScatterDimensionNumbersToString()
2414 StrJoin(scatter_dimension_numbers().inserted_window_dims(), ","), "}"); in ScatterDimensionNumbersToString()
2417 StrJoin(scatter_dimension_numbers().scatter_dims_to_operand_dims(), ","), in ScatterDimensionNumbersToString()
2420 "index_vector_dim=", scatter_dimension_numbers().index_vector_dim()); in ScatterDimensionNumbersToString()
2451 *proto.mutable_scatter_dimension_numbers() = scatter_dimension_numbers(); in ToProto()
2466 scatter_dimension_numbers(), in IdenticalSlowPath()
2467 casted_other.scatter_dimension_numbers()) && in IdenticalSlowPath()
2477 scatter_dimension_numbers()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 542 auto scatter_dimension_numbers = in CreateFromProto() local
544 proto.scatter_dimension_numbers()); in CreateFromProto()
546 computations(0), *scatter_dimension_numbers); in CreateFromProto()
3524 const ScatterDimensionNumbers& HloInstruction::scatter_dimension_numbers() in scatter_dimension_numbers() function in xla::HloInstruction
3526 return Cast<HloScatterInstruction>(this)->scatter_dimension_numbers(); in scatter_dimension_numbers()
|
| D | hlo.proto | 185 xla.ScatterDimensionNumbers scatter_dimension_numbers = 48; field
|
| D | hlo_evaluator_typed_visitor.h | 2309 scatter->scatter_dimension_numbers(); in HandleScatter()
2334 &scatter->scatter_dimension_numbers(), /*input_shape=*/operand_shape, in HandleScatter()
2337 scatter->scatter_dimension_numbers(), /*input_shape=*/operand_shape, in HandleScatter()
|
| D | hlo_instruction.h | 1598 const ScatterDimensionNumbers& scatter_dimension_numbers() const;
|
| D | hlo_verifier.cc | 824 scatter->scatter_dimension_numbers())); in HandleScatter()
|
| D | hlo_instructions.h | 1406 const ScatterDimensionNumbers& scatter_dimension_numbers() const { in scatter_dimension_numbers() function
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 474 : use->scatter_dimension_numbers() in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1082 scatter->scatter_dimension_numbers(); in EmitScatter()
|