Home
last modified time | relevance | path

Searched refs:kParameter (Results 1 – 25 of 110) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc331 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/v8/src/compiler/
Djs-context-specialization.cc22 case IrOpcode::kParameter: in Reduce()
37 DCHECK_EQ(IrOpcode::kParameter, node->opcode()); in ReduceParameter()
91 DCHECK_EQ(IrOpcode::kParameter, node->opcode()); in IsContextParameter()
114 case IrOpcode::kParameter: { in GetSpecializationContext()
244 case IrOpcode::kParameter: { in GetModuleContext()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc38 if (hlo.opcode() == HloOpcode::kParameter && in AddAliasingInformationToIrArray()
81 if (hlo.opcode() == HloOpcode::kParameter && in AddAliasingInformationToIrArray()
142 if (instruction->opcode() == HloOpcode::kParameter) { in GetNoaliasMetadataForBuffer()
Dfused_ir_emitter.cc119 if (tuple_operand->opcode() != HloOpcode::kParameter) { in HandleGetTupleElement()
237 if (instruction->opcode() == HloOpcode::kParameter) { in IsFusedIrEmitterInefficient()
Ddynamic_update_slice_util.cc51 HloOpcode::kParameter; in MayBeImplementedAsInPlaceDynamicUpdateSlice()
82 CHECK_EQ(fusion_operand->opcode(), HloOpcode::kParameter); in CanEmitFusedDynamicUpdateSliceInPlace()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_merger.cc57 CHECK_EQ(HloOpcode::kParameter, param->opcode()); in CalculateBytesReadByFusionParameter()
91 if (fused_instruction->opcode() != HloOpcode::kParameter) { in CalculateBytesReadByFusionInstruction()
222 return instruction->opcode() != HloOpcode::kParameter && in HandleFusion()
Dalias_passthrough_params.cc33 if (root->operand(i)->opcode() == HloOpcode::kParameter && in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dfusion_node_indexing_evaluation.cc87 if (instruction->opcode() == HloOpcode::kParameter) { in RecomputeCache()
117 if (operand->opcode() == HloOpcode::kParameter) { in UpdateIndexingUsersOfOperands()
Dhlo_computation.cc55 if (instruction->opcode() == HloOpcode::kParameter) { in Build()
78 if (instruction->opcode() == HloOpcode::kParameter) { in HloComputation()
97 CHECK(instruction->opcode() != HloOpcode::kParameter) in AddInstruction()
118 CHECK(instruction->opcode() == HloOpcode::kParameter); in AddParameter()
129 CHECK_EQ(instruction->opcode(), HloOpcode::kParameter); in AddEntryComputationParameter()
150 CHECK_EQ(instruction->opcode(), HloOpcode::kParameter); in ReplaceEntryComputationParameter()
234 if (instruction->opcode() == HloOpcode::kParameter && in IsSafelyRemovable()
655 if (instruction->opcode() == HloOpcode::kParameter) { in CreateFromProto()
678 if (instruction->opcode() == HloOpcode::kParameter) { in CreateFromProto()
1062 CHECK_EQ(instr->opcode(), HloOpcode::kParameter) in CloneWithReplacements()
[all …]
Dhlo_query.cc39 if (operand->opcode() != HloOpcode::kParameter && in AllOperandsAreParametersOrConstants()
49 if (operand->opcode() != HloOpcode::kParameter) { in AllOperandsAreParameters()
Dar_crs_combiner.cc101 CHECK_EQ(HloOpcode::kParameter, instruction->opcode()); in WhileFromBodyParameter()
115 CHECK_EQ(HloOpcode::kParameter, instruction->opcode()); in ConditionalFromBodyParameter()
142 case HloOpcode::kParameter: { in GetAllTuples()
289 if (opcode1 == HloOpcode::kParameter) { in InstructionsComputeSameValue()
Dhlo_ordering.cc105 b.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
110 a.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
131 v.defining_instruction()->opcode() == HloOpcode::kParameter; in IsDefinedBefore()
Dhlo_graph_dumper.cc542 if (to_node->IsFused() && to_node->opcode() == HloOpcode::kParameter) { in Header()
725 if (instr->opcode() != HloOpcode::kParameter || !instr->IsFused()) { in TryGetFusionParameterConstant()
753 return instr->opcode() == HloOpcode::kParameter && instr->shape().IsTuple() && in ShouldMergeIntoUsers()
878 if (operand->opcode() == HloOpcode::kParameter) { in GetInstructionNodeInlinedOperands()
925 return operand->opcode() == HloOpcode::kParameter && in GetInstructionColor()
1042 case HloOpcode::kParameter: in GetInstructionColor()
1091 if (instr->opcode() == HloOpcode::kParameter) { in GetInstructionNodeLabel()
1226 if (instr->opcode() == HloOpcode::kParameter && instr->IsFused()) { in AddInstructionIncomingEdges()
Dhlo_live_range.cc164 if (value->instruction()->opcode() == HloOpcode::kParameter) { in CalculateBufferStartEndMap()
193 if (value->instruction()->opcode() == HloOpcode::kParameter && in CalculateBufferStartEndMap()
Ddefuser.cc67 if (fused_instruction->opcode() == HloOpcode::kParameter) { in Defuse()
Dhlo_constant_folding.cc95 if (instruction->opcode() == HloOpcode::kParameter || in Run()
Dhlo_liveness_analysis.cc198 CHECK_EQ(instruction->opcode(), HloOpcode::kParameter); in PropagateLivenessToParameterCallers()
291 } else if (instruction->opcode() == HloOpcode::kParameter) { in RunAnalysis()
Dmap_inliner.cc79 if (root.opcode() == HloOpcode::kParameter) { in HandleMap()
Dhlo_casting_utils_test.cc33 : HloInstruction(HloOpcode::kParameter, ShapeUtil::MakeShape(F32, {})) {} in AnotherDummyInstruction()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc206 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
209 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
256 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
259 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
394 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
446 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
489 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
529 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
532 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
570 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_select_and_scatter_op.cc68 select_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
89 scatter_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
Dgather_scatter_ops.cc72 update_computation_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
Dxla_reduce_op.cc61 reducer_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
/external/v8/src/torque/
Dinstructions.h83 kParameter, enumerator
91 return DefinitionLocation(Kind::kParameter, nullptr, index); in Parameter()
105 bool IsParameter() const { return kind_ == Kind::kParameter; } in IsParameter()
172 case DefinitionLocation::Kind::kParameter:
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache_test.cc64 : XlaCompiler::Argument::kParameter); in BM_BuildSignature()

12345