Home
last modified time | relevance | path

Searched refs:parameter_number (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h270 int64 parameter_number, const Literal& literal, const string& name,
277 int64 parameter_number, const Literal& literal, const string& name,
328 NativeT value, int64 parameter_number, const string& name,
341 tensorflow::gtl::ArraySlice<NativeT> values, int64 parameter_number,
356 const Array2D<NativeT>& array_2d, int64 parameter_number,
371 const Array3D<NativeT>& array_3d, int64 parameter_number,
549 NativeT value, int64 parameter_number, const string& name, in CreateR0Parameter() argument
557 *data_handle = builder->Parameter(parameter_number, literal->shape(), name); in CreateR0Parameter()
563 tensorflow::gtl::ArraySlice<NativeT> values, int64 parameter_number, in CreateR1Parameter() argument
572 *data_handle = builder->Parameter(parameter_number, literal->shape(), name); in CreateR1Parameter()
[all …]
Dclient_library_test_base.cc527 int64 parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral() argument
529 return CreateParameterAndTransferLiteral(parameter_number, literal, name, in CreateParameterAndTransferLiteral()
535 int64 parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral() argument
548 builder->Parameter(parameter_number, param_literal->shape(), name); in CreateParameterAndTransferLiteral()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.h44 explicit HloParameterMatcher(int64 parameter_number) in HloParameterMatcher() argument
46 parameter_number_(parameter_number) {}
177 int64 parameter_number) { in Parameter() argument
179 new ::xla::testing::HloParameterMatcher(parameter_number)); in Parameter()
Dsource_map_util.cc47 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument() argument
54 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
Dhlo_matchers.cc82 if (instruction->parameter_number() != parameter_number_) { in MatchAndExplain()
84 << instruction->parameter_number() << ", want " in MatchAndExplain()
Dbuffer_assignment.h89 int64 parameter_number() const { in parameter_number() function
213 void set_entry_computation_parameter(int64 parameter_number, in set_entry_computation_parameter() argument
216 parameter_number_ = parameter_number; in set_entry_computation_parameter()
Dsource_map_util.h33 Status InvalidParameterArgument(Executable* executable, int parameter_number,
Dhlo_graph_dumper.cc258 auto n0 = operand0->parameter_number(); in MatchTrivialComputation()
259 auto n1 = operand1->parameter_number(); in MatchTrivialComputation()
813 const HloInstruction* operand = fusion->operand(instr->parameter_number()); in GetInstructionNodeInlinedOperands()
831 operand_str = Printf("Parameter %lld", operand->parameter_number()); in GetInstructionNodeInlinedOperands()
1013 return Printf("<b>Parameter %lld</b>", instr->parameter_number()); in GetInstructionNodeLabel()
1141 add_edge(fusion->operand(instr->parameter_number()), instr, in AddInstructionIncomingEdges()
Dhlo_module.cc248 TF_RET_CHECK(!ContainsKey(parameters, instruction.parameter_number())) in ProgramShapeFromProto()
251 << instruction.parameter_number(); in ProgramShapeFromProto()
252 parameters[instruction.parameter_number()] = {instruction.name(), in ProgramShapeFromProto()
Dliveness_util.cc47 return user->operand(fused_param->parameter_number()) == operand; in DoesNotUseOperandBuffer()
151 return fusion->operand(fused_param->parameter_number()) == operand; in HasUniqueFusedUseOfOperandAt()
Dinliner.cc87 params.push_back(map->operands()[root.operand(o)->parameter_number()]); in HandleMap()
Dcall_inliner.cc71 parameter, call_->mutable_operand(parameter->parameter_number()))); in HandleParameter()
Dhlo_dataflow_analysis.cc445 callsite.instruction()->operand(parameter->parameter_number()))); in UpdateParameterValueSet()
450 CHECK_EQ(parameter->parameter_number(), 0); in UpdateParameterValueSet()
463 CHECK_EQ(parameter->parameter_number(), 0); in UpdateParameterValueSet()
Dhlo.proto57 int64 parameter_number = 9; field
206 int64 parameter_number = 6; field
Dhlo_instruction.h198 static std::unique_ptr<HloInstruction> CreateParameter(int64 parameter_number,
693 int64 parameter_number() const { in parameter_number() function
895 HloInstruction* fused_parameter(int64 parameter_number) const;
Duser_computation.h348 int parameter_number) const;
Dhlo_instruction.cc107 instruction->parameter_number_ = proto.parameter_number(); in CreateFromProto()
157 int64 parameter_number, const Shape& shape, const string& name) { in CreateParameter() argument
160 instruction->parameter_number_ = parameter_number; in CreateParameter()
839 clone->mutable_operand(fused_instruction->parameter_number()))); in MergeFusionInstruction()
879 fused_instruction->parameter_number())); in MergeFusionInstructionIntoMultiOutput()
1743 return parameter_number() == other.parameter_number() && in IdenticalSlowPath()
2478 HloInstruction* HloInstruction::fused_parameter(int64 parameter_number) const { in fused_parameter()
2481 parameter_number); in fused_parameter()
Dhlo_tfgraph_builder.cc125 StrAppend(&instruction_name, ".", instruction->parameter_number()); in GetNodeNameForInstruction()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc121 return parameter_arrays_[parameter->parameter_number()] in HandleParameter()
128 parameter_arrays_[parameter->parameter_number()].GetBasePointer())); in HandleParameter()
Dops.h57 auto* operand = fusion->operand(fusion_operand->parameter_number()); in CanEmitFusedDynamicUpdateSliceInPlace()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc308 fusion_instruction->operand(fused_dot->operand(0)->parameter_number()); in RunDotOutputFusion()
310 fusion_instruction->operand(fused_dot->operand(1)->parameter_number()); in RunDotOutputFusion()
312 fused_add->operand(1 - dot_operand_idx_in_add)->parameter_number()); in RunDotOutputFusion()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer.cc312 gte_fusion_param0->parameter_number()) != in MatchExprTree()
473 inst->parameter_number()) != in MatchExprTree()
/external/syslinux/com32/include/
Dmenu.h95 enum parameter_number { enum
/external/clang/test/SemaCXX/
Daddr-of-overloaded-function.cpp182 void parameter_number() { in parameter_number() function
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc301 ComputationDataHandle LocalComputationBuilder::Parameter(int64 parameter_number, in Parameter() argument
304 return builder_.Parameter(parameter_number, shape, name); in Parameter()

12