Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc329 HloOpcode::kParameter}); in TEST_F()
361 HloOpcode::kBroadcast, HloOpcode::kParameter, HloOpcode::kParameter, in TEST_F()
362 HloOpcode::kParameter}); in TEST_F()
381 {HloOpcode::kNegate, HloOpcode::kBroadcast, HloOpcode::kParameter}); in TEST_F()
403 HloOpcode::kParameter, HloOpcode::kParameter}); in TEST_F()
421 {HloOpcode::kNegate, HloOpcode::kExp, HloOpcode::kParameter}); in TEST_F()
440 {HloOpcode::kNegate, HloOpcode::kReshape, HloOpcode::kParameter}); in TEST_F()
458 {HloOpcode::kNegate, HloOpcode::kReverse, HloOpcode::kParameter}); in TEST_F()
477 {HloOpcode::kNegate, HloOpcode::kSlice, HloOpcode::kParameter}); in TEST_F()
500 HloOpcode::kParameter}); in TEST_F()
[all …]
/external/v8/src/compiler/
Djs-context-specialization.cc21 case IrOpcode::kParameter: in Reduce()
34 DCHECK_EQ(IrOpcode::kParameter, node->opcode()); in ReduceParameter()
88 DCHECK_EQ(IrOpcode::kParameter, node->opcode()); in IsContextParameter()
111 case IrOpcode::kParameter: { in GetSpecializationContext()
Dbasic-block-instrumentor.cc30 case IrOpcode::kParameter: in FindInsertionPoint()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc37 if (hlo.opcode() == HloOpcode::kParameter && in AddAliasingInformationToIrArray()
81 if (hlo.opcode() == HloOpcode::kParameter) { in AddAliasingInformationToIrArray()
169 buffer->instruction()->opcode() == HloOpcode::kParameter) { in GetNoaliasMetadataForBuffer()
Dfused_ir_emitter.cc112 if (tuple_operand->opcode() != HloOpcode::kParameter) { in HandleGetTupleElement()
232 if (instruction->opcode() == HloOpcode::kParameter) { in IsFusedIrEmitterInefficient()
Ddynamic_update_slice_util.h57 if (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()
228 return instruction->opcode() != HloOpcode::kParameter && in HandleFusion()
Dinstruction_fusion.cc140 if (op1->opcode() != HloOpcode::kParameter) { in ShouldFuseInexpensiveChecks()
143 if (op1->opcode() != HloOpcode::kParameter || in ShouldFuseInexpensiveChecks()
Dstream_assignment.cc69 if (hlo.opcode() == HloOpcode::kParameter || in ComputeStreamToAssign()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_query.cc37 if (operand->opcode() != HloOpcode::kParameter && in AllOperandsAreParametersOrConstants()
47 if (operand->opcode() != HloOpcode::kParameter) { in AllOperandsAreParameters()
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()
204 if (instruction->opcode() == HloOpcode::kParameter && in IsRemovable()
547 if (instruction->opcode() == HloOpcode::kParameter) { in CreateFromProto()
570 if (instruction->opcode() == HloOpcode::kParameter) { in CreateFromProto()
999 CHECK_EQ(instr->opcode(), HloOpcode::kParameter) in CloneWithReplacements()
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()
Dar_crs_combiner.cc97 CHECK_EQ(HloOpcode::kParameter, instruction->opcode()); in WhileFromBodyParameter()
117 if (instruction->opcode() == HloOpcode::kParameter) { in GetAllTuples()
214 if (opcode1 == HloOpcode::kParameter) { in InstructionsComputeSameValue()
Dhlo_graph_dumper.cc538 if (to_node->IsFused() && to_node->opcode() == HloOpcode::kParameter) { in Header()
719 if (instr->opcode() != HloOpcode::kParameter || !instr->IsFused()) { in TryGetFusionParameterConstant()
747 return instr->opcode() == HloOpcode::kParameter && instr->shape().IsTuple() && in ShouldMergeIntoUsers()
866 if (operand->opcode() == HloOpcode::kParameter) { in GetInstructionNodeInlinedOperands()
913 return operand->opcode() == HloOpcode::kParameter && in GetInstructionColor()
1026 case HloOpcode::kParameter: in GetInstructionColor()
1073 if (instr->opcode() == HloOpcode::kParameter) { in GetInstructionNodeLabel()
1207 if (instr->opcode() == HloOpcode::kParameter && instr->IsFused()) { in AddInstructionIncomingEdges()
Dhlo_liveness_analysis.cc198 CHECK_EQ(instruction->opcode(), HloOpcode::kParameter); in PropagateLivenessToParameterCallers()
291 } else if (instruction->opcode() == HloOpcode::kParameter) { in RunAnalysis()
Dhlo_constant_folding.cc95 if (instruction->opcode() == HloOpcode::kParameter || in Run()
Ddefuser.cc67 if (fused_instruction->opcode() == HloOpcode::kParameter) { in Defuse()
Dmap_inliner.cc79 if (root.opcode() == HloOpcode::kParameter) { in HandleMap()
Dbfloat16_support.cc115 if (inst->opcode() == HloOpcode::kParameter) { in EffectiveOperandPrecisionIsOutputPrecision()
Dhlo_casting_utils_test.cc33 : HloInstruction(HloOpcode::kParameter, ShapeUtil::MakeShape(F32, {})) {} in AnotherDummyInstruction()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc198 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
201 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
248 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
251 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
301 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
352 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
395 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
435 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
438 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
476 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_select_and_scatter_op.cc69 select_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
90 scatter_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
Dxla_reduce_op.cc61 reducer_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
/external/v8/src/torque/
Ddeclarable.h30 kParameter, enumerator
48 bool IsParameter() const { return kind() == kParameter; } in IsParameter()
116 : Value(Declarable::kParameter, type, name), var_name_(var_name) {} in Parameter()
/external/v8/src/objects/
Dmanaged.h103 v8::WeakCallbackType::kParameter); in FromSharedPtr()

1234