| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fake_quant_num_bits.cc | 106 case OperatorType::kGather: in DoesOpBlockBackwardPropagation()
132 case OperatorType::kGather: in DoesOpInputBlockBackwardPropagation()
|
| D | resolve_gather_attributes.cc | 32 if (gather_op->type != OperatorType::kGather) in Run()
|
| D | propagate_activation_function_into_constants.cc | 57 case OperatorType::kGather: in Run()
|
| D | resolve_constant_gather.cc | 73 if (base_op->type != OperatorType::kGather) { in Run()
|
| D | unpartition_embedding_lookup.cc | 118 if (op->type != OperatorType::kGather) { in Run()
|
| D | hardcode_min_max.cc | 440 case OperatorType::kGather: in Run()
|
| D | quantize.cc | 58 type == OperatorType::kLstmCell || type == OperatorType::kGather || in SupportsQuantization()
|
| D | propagate_fixed_sizes.cc | 2117 case OperatorType::kGather: in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 404 } else if ((opcode == HloOpcode::kGather || in FindConstrainedUses()
467 case HloOpcode::kGather: in CreateLiteralForConstrainedUses()
472 use->opcode() == HloOpcode::kGather in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 38 hlo.opcode() == HloOpcode::kGather || in CanBeLoopFused()
|
| D | cpu_instruction_fusion_test.cc | 758 {HloOpcode::kGather, HloOpcode::kAdd, HloOpcode::kBroadcast, in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 90 V(kGather, "gather", 2) \
|
| D | gather_expander.cc | 373 return inst->opcode() == HloOpcode::kGather && in InstructionMatchesPattern()
|
| D | instruction_fusion.cc | 138 case HloOpcode::kGather: in IsExpensive()
|
| D | hlo_instruction.cc | 522 case HloOpcode::kGather: { in CreateFromProto()
1384 case HloOpcode::kGather: in CloneWithNewOperands()
1835 case HloOpcode::kGather: in IdenticalSlowPath()
2649 case HloOpcode::kGather: in Visit()
|
| D | indexed_array_analysis.cc | 152 } else if (instr->opcode() == HloOpcode::kGather) { in ComputeArrayFor()
|
| D | hlo_graph_dumper.cc | 992 case HloOpcode::kGather: in GetInstructionColor()
|
| D | layout_assignment.cc | 2058 case HloOpcode::kGather: in InstructionCanChangeLayout()
|
| D | hlo_instructions.cc | 2313 : HloInstruction(HloOpcode::kGather, shape) { in HloGatherInstruction()
|
| D | elemental_ir_emitter.cc | 2380 case HloOpcode::kGather: in MakeElementGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 171 instr.opcode() == HloOpcode::kGather || in IsLoopFusible()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 85 kGather, enumerator
1727 GatherOperator() : Operator(OperatorType::kGather) {}
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 59 constexpr char kGather[] = "Gather"; variable
256 {kGather, wrap(&OpLevelCostEstimator::PredictGatherOrSlice)}, in OpLevelCostEstimator()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 241 SerializeAndDeserialize(GetOperator("GATHER", OperatorType::kGather), op); in TEST_F()
|
| D | operator.cc | 2369 OperatorType::kGather)); in BuildOperatorList()
|