Home
last modified time | relevance | path

Searched refs:param_no (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.h51 const ShapeLayout& parameter_layout(int64_t param_no) const { in parameter_layout() argument
52 return parameter_layouts_[param_no]; in parameter_layout()
54 ShapeLayout* mutable_parameter_layout(int64_t param_no) { in mutable_parameter_layout() argument
55 return &parameter_layouts_[param_no]; in mutable_parameter_layout()
72 const Shape& parameter_shape(int64_t param_no) const { in parameter_shape() argument
73 return parameter_layouts_[param_no].shape(); in parameter_shape()
Dhlo_computation.cc90 int64_t param_no = instruction->parameter_number(); in HloComputation() local
91 CHECK(param_no >= 0 && param_no < parameter_count) in HloComputation()
93 << parameter_count << "), got " << param_no; in HloComputation()
94 CHECK(param_instructions_[param_no] == nullptr) in HloComputation()
95 << "\nERROR: parameter number " << param_no in HloComputation()
97 param_instructions_[param_no] = instruction.get(); in HloComputation()
183 int64_t param_no, HloInstruction* old_instruction, in ReplaceEntryComputationParameter() argument
185 CHECK_GE(param_no, 0); in ReplaceEntryComputationParameter()
186 CHECK_LT(param_no, param_instructions_.size()); in ReplaceEntryComputationParameter()
192 param_no) = ShapeLayout(instruction->shape()); in ReplaceEntryComputationParameter()
[all …]
Dhlo_computation.h152 HloInstruction* ReplaceParameter(int64_t param_no,
158 Status RemoveParameter(int64_t param_no);
187 int64_t param_no, HloInstruction* old_instruction,
223 HloInstruction* parameter_instruction(int64_t param_no) const { in parameter_instruction() argument
224 CHECK_GE(param_no, 0); in parameter_instruction()
225 CHECK_LT(param_no, static_cast<int64_t>(param_instructions_.size())) in parameter_instruction()
226 << "Computation " << name() << " has no parameter number " << param_no; in parameter_instruction()
227 return param_instructions_[param_no]; in parameter_instruction()
Dhlo_verifier.cc1141 int64_t param_no = fused_param->parameter_number(); in HandleFusion() local
1142 if (!ShapesSame(fused_param->shape(), fusion->operand(param_no)->shape())) { in HandleFusion()
1146 param_no, fusion->ToString().c_str()); in HandleFusion()
2265 int64_t param_no = fused_param->parameter_number(); in CheckFusionInstruction() local
2266 if (param_no < 0) { in CheckFusionInstruction()
2268 param_no, fusion->ToString()); in CheckFusionInstruction()
2270 if (param_no >= fused_parameters.size()) { in CheckFusionInstruction()
2274 param_no, fusion->ToString(), fused_parameters.size()); in CheckFusionInstruction()
2276 if (parameter_numbers[param_no]) { in CheckFusionInstruction()
2278 "Did not expect parameter number %d more than once in %s.", param_no, in CheckFusionInstruction()
[all …]
Dhlo_instructions.cc1599 const int64_t param_no = operand_count(); in AddCallOperand() local
1600 std::string param_name = StrCat("param_", param_no); in AddCallOperand()
1603 param_no, new_operand->shape(), param_name)); in AddCallOperand()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h300 void ForceParameterLayout(HloModule* module, int64_t param_no, in ForceParameterLayout() argument
302 ASSERT_LT(param_no, in ForceParameterLayout()
305 ->mutable_parameter_layout(param_no) in ForceParameterLayout()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc466 int64_t param_no = allocation.parameter_number(); in BufferForAllocation() local
470 return (*unowned_shapedbuffers)[param_no]->buffers().element( in BufferForAllocation()
473 return std::get<absl::Span<ExecutionInput>>(arguments)[param_no] in BufferForAllocation()
484 allocation.param_shape_index().ToString(), param_no); in BufferForAllocation()