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 param_no) const { in parameter_layout() argument
52 return parameter_layouts_[param_no]; in parameter_layout()
54 ShapeLayout* mutable_parameter_layout(int64 param_no) { in mutable_parameter_layout() argument
55 return &parameter_layouts_[param_no]; in mutable_parameter_layout()
72 const Shape& parameter_shape(int64 param_no) const { in parameter_shape() argument
73 return parameter_layouts_[param_no].shape(); in parameter_shape()
Dhlo_computation.cc79 int64 param_no = instruction->parameter_number(); in HloComputation() local
80 CHECK(param_no >= 0 && param_no < parameter_count) in HloComputation()
82 << parameter_count << "), got " << param_no; in HloComputation()
83 CHECK(param_instructions_[param_no] == nullptr) in HloComputation()
84 << "\nERROR: parameter number " << param_no in HloComputation()
86 param_instructions_[param_no] = instruction.get(); in HloComputation()
145 Status HloComputation::RemoveParameter(int64 param_no) { in RemoveParameter() argument
146 CHECK_GE(param_no, 0); in RemoveParameter()
147 CHECK_LT(param_no, param_instructions_.size()); in RemoveParameter()
149 HloInstruction* param_instruction = param_instructions_[param_no]; in RemoveParameter()
[all …]
Dhlo_computation.h113 Status RemoveParameter(int64 param_no);
159 HloInstruction* parameter_instruction(int64 param_no) const { in parameter_instruction() argument
160 CHECK_GE(param_no, 0); in parameter_instruction()
161 CHECK_LT(param_no, static_cast<int64>(param_instructions_.size())) in parameter_instruction()
162 << "Computation " << name() << " has no parameter number " << param_no; in parameter_instruction()
163 return param_instructions_[param_no]; in parameter_instruction()
Dhlo_verifier.cc523 int64 param_no = fused_param->parameter_number(); in HandleFusion() local
524 if (!ShapesSame(fused_param->shape(), fusion->operand(param_no)->shape())) { in HandleFusion()
528 param_no, fusion->ToString().c_str()); in HandleFusion()
1256 int64 param_no = fused_param->parameter_number(); in CheckFusionInstruction() local
1257 if (param_no < 0) { in CheckFusionInstruction()
1259 param_no, fusion->ToString()); in CheckFusionInstruction()
1261 if (param_no >= fused_parameters.size()) { in CheckFusionInstruction()
1265 param_no, fusion->ToString(), fused_parameters.size()); in CheckFusionInstruction()
1267 if (parameter_numbers[param_no]) { in CheckFusionInstruction()
1269 "Did not expect parameter number %d more than once in %s.", param_no, in CheckFusionInstruction()
[all …]
Dhlo_instructions.cc1218 const int64 param_no = operand_count(); in AddFusionOperand() local
1222 string param_name = StrCat("param_", param_no); in AddFusionOperand()
1225 HloInstruction::CreateParameter(param_no, new_operand->shape(), in AddFusionOperand()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h264 void ForceParameterLayout(HloModule* module, int64 param_no, in ForceParameterLayout() argument
266 ASSERT_LT(param_no, in ForceParameterLayout()
269 ->mutable_parameter_layout(param_no) in ForceParameterLayout()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc255 auto param_no = allocation.parameter_number(); in ExecuteOnStream() local
257 arguments[param_no]->buffer(allocation.param_shape_index()); in ExecuteOnStream()
266 allocation.param_shape_index().ToString(), param_no); in ExecuteOnStream()