| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 331 HloOpcode::kParameter}); in TEST_F()
363 HloOpcode::kBroadcast, HloOpcode::kParameter, HloOpcode::kParameter, in TEST_F()
364 HloOpcode::kParameter}); in TEST_F()
383 {HloOpcode::kNegate, HloOpcode::kBroadcast, HloOpcode::kParameter}); in TEST_F()
405 HloOpcode::kParameter, HloOpcode::kParameter}); in TEST_F()
423 {HloOpcode::kNegate, HloOpcode::kExp, HloOpcode::kParameter}); in TEST_F()
442 {HloOpcode::kNegate, HloOpcode::kReshape, HloOpcode::kParameter}); in TEST_F()
460 {HloOpcode::kNegate, HloOpcode::kReverse, HloOpcode::kParameter}); in TEST_F()
479 {HloOpcode::kNegate, HloOpcode::kSlice, HloOpcode::kParameter}); in TEST_F()
502 HloOpcode::kParameter}); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | alias_analysis.cc | 38 if (hlo.opcode() == HloOpcode::kParameter && in AddAliasingInformationToIrArray()
81 if (hlo.opcode() == HloOpcode::kParameter && in AddAliasingInformationToIrArray()
142 if (instruction->opcode() == HloOpcode::kParameter) { in GetNoaliasMetadataForBuffer()
|
| D | dynamic_update_slice_util.cc | 51 HloOpcode::kParameter; in MayBeImplementedAsInPlaceDynamicUpdateSlice()
82 CHECK_EQ(fusion_operand->opcode(), HloOpcode::kParameter); in CanEmitFusedDynamicUpdateSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_computation.cc | 55 if (instruction->opcode() == HloOpcode::kParameter) { in Build()
78 if (instruction->opcode() == HloOpcode::kParameter) { in HloComputation()
97 CHECK(instruction->opcode() != HloOpcode::kParameter) in AddInstruction()
121 CHECK(instruction->opcode() == HloOpcode::kParameter); in AddParameter()
132 CHECK_EQ(instruction->opcode(), HloOpcode::kParameter); in AddEntryComputationParameter()
153 CHECK_EQ(instruction->opcode(), HloOpcode::kParameter); in ReplaceEntryComputationParameter()
237 if (instruction->opcode() == HloOpcode::kParameter && in IsSafelyRemovable()
675 if (instruction->opcode() == HloOpcode::kParameter) { in CreateFromProto()
698 if (instruction->opcode() == HloOpcode::kParameter) { in CreateFromProto()
1096 CHECK_EQ(instr->opcode(), HloOpcode::kParameter) in CloneWithReplacements()
[all …]
|
| D | fusion_node_indexing_evaluation.cc | 134 if (instruction->opcode() == HloOpcode::kParameter) { in RecomputeCache()
154 if (operand->opcode() == HloOpcode::kParameter) { in UpdateIndexingUsersOfOperands()
|
| D | hlo_query.cc | 39 if (operand->opcode() != HloOpcode::kParameter && in AllOperandsAreParametersOrConstants()
49 if (operand->opcode() != HloOpcode::kParameter) { in AllOperandsAreParameters()
|
| D | hlo_live_range.cc | 164 if (value->instruction()->opcode() == HloOpcode::kParameter) { in CalculateBufferStartEndMap()
193 if (value->instruction()->opcode() == HloOpcode::kParameter && in CalculateBufferStartEndMap()
|
| D | hlo_ordering.cc | 147 b.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
152 a.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
173 v.defining_instruction()->opcode() == HloOpcode::kParameter; in IsDefinedBefore()
|
| D | ar_crs_combiner.cc | 194 CHECK_EQ(HloOpcode::kParameter, instruction->opcode()); in WhileFromBodyParameter()
208 CHECK_EQ(HloOpcode::kParameter, instruction->opcode()); in ConditionalFromBodyParameter()
235 case HloOpcode::kParameter: { in GetAllTuples()
385 if (opcode1 == HloOpcode::kParameter) { in InstructionsComputeSameValue()
|
| D | hlo_graph_dumper.cc | 544 if (to_node->IsFused() && to_node->opcode() == HloOpcode::kParameter) { in Header()
728 if (instr->opcode() != HloOpcode::kParameter || !instr->IsFused()) { in TryGetFusionParameterConstant()
756 return instr->opcode() == HloOpcode::kParameter && instr->shape().IsTuple() && in ShouldMergeIntoUsers()
881 if (operand->opcode() == HloOpcode::kParameter) { in GetInstructionNodeInlinedOperands()
928 return operand->opcode() == HloOpcode::kParameter && in GetInstructionColor()
1049 case HloOpcode::kParameter: in GetInstructionColor()
1101 if (instr->opcode() == HloOpcode::kParameter) { in GetInstructionNodeLabel()
1246 if (instr->opcode() == HloOpcode::kParameter && instr->IsFused()) { in AddInstructionIncomingEdges()
|
| D | hlo_constant_folding.cc | 95 if (instruction->opcode() == HloOpcode::kParameter || in Run()
|
| D | defuser.cc | 67 if (fused_instruction->opcode() == HloOpcode::kParameter) { in Defuse()
|
| D | while_loop_all_reduce_code_motion.cc | 125 case HloOpcode::kParameter: { in IsAllReduceMovable()
132 if (instruction->opcode() == HloOpcode::kParameter) { in IsAllReduceMovable()
207 return instruction->opcode() == HloOpcode::kParameter; in IsAllReduceMovable()
|
| D | map_inliner.cc | 79 if (root.opcode() == HloOpcode::kParameter) { in HandleMap()
|
| D | hlo_liveness_analysis.cc | 198 CHECK_EQ(instruction->opcode(), HloOpcode::kParameter); in PropagateLivenessToParameterCallers()
291 } else if (instruction->opcode() == HloOpcode::kParameter) { in RunAnalysis()
|
| D | bfloat16_support.cc | 118 if (inst->opcode() == HloOpcode::kParameter) { in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | hlo_casting_utils_test.cc | 33 : HloInstruction(HloOpcode::kParameter, ShapeUtil::MakeShape(F32, {})) {} in AnotherDummyInstruction()
|
| D | memory_space_propagation.cc | 90 if (instruction->opcode() == HloOpcode::kParameter && in Propagate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fusion_merger.cc | 58 CHECK_EQ(HloOpcode::kParameter, param->opcode()); in CalculateBytesReadByFusionParameter()
92 if (fused_instruction->opcode() != HloOpcode::kParameter) { in CalculateBytesReadByFusionInstruction()
256 return instruction->opcode() != HloOpcode::kParameter && in HandleFusion()
|
| D | alias_passthrough_params.cc | 33 if (root->operand(i)->opcode() == HloOpcode::kParameter && in Run()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 206 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
209 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
255 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
258 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
392 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
445 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
508 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
548 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
551 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
589 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
[all …]
|
| D | xla_argument.h | 47 kParameter, enumerator
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_select_and_scatter_op.cc | 68 select_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
89 scatter_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
|
| D | gather_scatter_ops.cc | 73 update_computation_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
|
| /external/skia/src/sksl/ir/ |
| D | SkSLVariable.h | 32 kParameter enumerator
|