Home
last modified time | relevance | path

Searched refs:num_parameters (Results 1 – 25 of 39) sorted by relevance

12

/external/apache-commons-bcel/src/main/java/org/apache/bcel/classfile/
DParameterAnnotations.java46 final int num_parameters = input.readUnsignedByte(); in ParameterAnnotations() local
47 parameter_annotation_table = new ParameterAnnotationEntry[num_parameters]; in ParameterAnnotations()
48 for (int i = 0; i < num_parameters; i++) { in ParameterAnnotations()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc78 if (computation->num_parameters() != arguments.size()) { in ExecuteAsyncOnStream()
84 for (int64 i = 0; i < computation->num_parameters(); ++i) { in ExecuteAsyncOnStream()
111 for (int64 p = 0; p < computation->num_parameters(); ++p) { in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_parameter_binding.cc116 dynamic_parameter.parameter_num < entry->num_parameters()); in Verify()
117 TF_RET_CHECK(dynamic_dimension.parameter_num < entry->num_parameters()); in Verify()
Dhlo_verifier.cc71 if (computation->num_parameters() != expected) { in CheckParameterCount()
75 computation->num_parameters()); in CheckParameterCount()
565 for (int64 parameter_idx = 0; parameter_idx < compare->num_parameters(); in HandleSort()
785 for (int64 i = 0; i < call->to_apply()->num_parameters(); ++i) { in HandleCall()
1292 if (computation->num_parameters() != layout.parameter_count()) { in VerifyEntryComputationLayout()
1296 layout.parameter_count(), computation->num_parameters()); in VerifyEntryComputationLayout()
1299 for (int i = 0; i < computation->num_parameters(); ++i) { in VerifyEntryComputationLayout()
1400 for (int i = 0; i < module.entry_computation()->num_parameters(); ++i) { in VerifyEntryAndExitShapes()
1818 if (while_cond->num_parameters() != 1) { in HandleWhile()
1821 while_cond->num_parameters(), while_cond->ToString()); in HandleWhile()
[all …]
Dall_reduce_combiner.cc70 hlo->to_apply()->num_parameters() == 2 && in CombineAllReduces()
192 instruction->to_apply()->num_parameters() != 2) { in CreateComputationGroups()
231 instruction->to_apply()->num_parameters() != 2) { in CreateComputationGroups()
Doptimize_input_output_buffer_alias.cc92 if (module->entry_computation()->num_parameters() != 1) { in Run()
Dhlo_input_output_alias_config.cc183 for (int64 i = 0; i < entry->num_parameters(); ++i) { in Verify()
192 TF_RET_CHECK(entry->num_parameters() > alias.parameter_number); in Verify()
Dwhile_loop_simplifier.cc503 TF_RET_CHECK(while_cond->num_parameters() == 1); in TryRemoveConstantParams()
504 TF_RET_CHECK(while_body->num_parameters() == 1); in TryRemoveConstantParams()
866 TF_RET_CHECK(while_cond->num_parameters() == 1); in TryFlattenNestedTuples()
867 TF_RET_CHECK(while_body->num_parameters() == 1); in TryFlattenNestedTuples()
993 TF_RET_CHECK(while_cond->num_parameters() == 1); in TryMergeInductionVariables()
994 TF_RET_CHECK(while_body->num_parameters() == 1); in TryMergeInductionVariables()
Dall_reduce_simplifier.cc89 all_reduce->to_apply()->num_parameters() != 2) { in Run()
Dstable_sort_expander.cc158 for (int64 i = 0; i < comparator->num_parameters(); ++i) { in ExpandInstruction()
Dlayout_assignment.cc578 CHECK_EQ(1, body->num_parameters()); in AddMandatoryConstraints()
579 CHECK_EQ(1, condition->num_parameters()); in AddMandatoryConstraints()
628 TF_RET_CHECK(instruction->branch_computation(j)->num_parameters() == 1); in AddMandatoryConstraints()
677 TF_RET_CHECK(computation->num_parameters() == call->operand_count()); in CheckCallLayout()
678 for (int64 i = 0; i < computation->num_parameters(); ++i) { in CheckCallLayout()
2145 entry->num_parameters()); in Run()
2146 for (int64 i = 0; i < entry->num_parameters(); ++i) { in Run()
Dhlo_replication_analysis.cc332 for (int i = 0; i < entry->num_parameters(); ++i) { in ComputeHloReplication()
Dhlo_computation.h193 int64 num_parameters() const { return param_instructions_.size(); } in num_parameters() function
Dcopy_insertion.cc358 entry->num_parameters()); in AddCopiesForAliasedInputOutputs()
379 TF_RET_CHECK(param->parameter_number() < entry->num_parameters()); in AddCopiesForAliasedInputOutputs()
Dwhile_loop_all_reduce_code_motion.cc70 if (all_reduce->to_apply()->num_parameters() != 2) { in IsAllReduceMovable()
Dhlo_evaluator.cc239 if (arg_literals.size() != computation.num_parameters()) { in Evaluate()
241 "Expected %d argument%s, but got %d.", computation.num_parameters(), in Evaluate()
242 computation.num_parameters() == 1 ? "" : "s", arg_literals.size()); in Evaluate()
2283 computation->num_parameters() == 2) { in IsScalarAdd()
/external/mesa3d/src/compiler/glsl/
Dopt_function_inlining.cpp151 unsigned num_parameters; in generate_inline() local
157 num_parameters = this->callee->parameters.length(); in generate_inline()
158 parameters = new ir_variable *[num_parameters]; in generate_inline()
Dserialize.cpp1055 uint32_t num_parameters = blob_read_uint32(metadata); in read_shader_parameters() local
1057 _mesa_reserve_parameter_storage(params, num_parameters); in read_shader_parameters()
1058 while (i < num_parameters) { in read_shader_parameters()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dalias_passthrough_params.cc26 if (module->entry_computation()->num_parameters() == 0 || in Run()
Dir_emitter.cc227 TF_RET_CHECK(nested_computation.num_parameters() > 0); in EmitCallToNestedComputation()
258 CHECK_EQ(2, computation.num_parameters()); in MaybeEmitDirectAtomicOperation()
500 if (computation.num_parameters() != 2) { in EmitAtomicOperationForNestedComputation()
505 computation.name(), computation.num_parameters()); in EmitAtomicOperationForNestedComputation()
Dir_emitter_unnested.cc1902 for (int i = 0; i < fused_computation->num_parameters(); i++) { in HandleFusion()
1946 for (int i = 0; i < fused_computation->num_parameters(); i++) { in HandleFusion()
2055 for (int i = 0; i < fused_computation->num_parameters(); i++) { in HandleFusion()
2765 CHECK_EQ(operand_shapes.size(), fused_computation->num_parameters()); in GetOrCreateSubComputationFromRegion()
2766 for (int i = 0; i < fused_computation->num_parameters(); i++) { in GetOrCreateSubComputationFromRegion()
3732 for (int i = 0; i < fused_computation->num_parameters(); i++) { in BuildFusedInitializerThunkForMlir()
5628 CHECK_LT(fused_computation->num_parameters(), ir_arrays.size()); in EmitReductionFromOrToContiguousDimensions()
5629 for (int i = 0; i < fused_computation->num_parameters(); i++) { in EmitReductionFromOrToContiguousDimensions()
5640 fused_computation->num_parameters(), num_reduces); in EmitReductionFromOrToContiguousDimensions()
5741 for (int i = 0; i < fused_computation->num_parameters(); i++) { in EmitElementForInputFusibleSlices()
[all …]
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc214 CHECK_EQ(computation->num_parameters(), 1); in GetParametersThatMustBeDonated()
220 return computation->num_parameters(); in GetParametersThatMustBeDonated()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc335 CHECK_EQ(entry_comp->num_parameters(), arguments.size()) in ExecuteAsyncOnStream()
337 for (int64 i = 0; i < entry_comp->num_parameters(); ++i) { in ExecuteAsyncOnStream()
/external/rust/crates/criterion/src/
Dbenchmark.rs485 let num_parameters = self.values.len(); in run() localVariable
504 let value_str = if num_parameters == 1 { in run()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc173 const int num_parameters = computation.num_parameters(); in ImportInstructionsImpl() local
175 if (arguments.size() != num_parameters) in ImportInstructionsImpl()
178 for (int i = 0; i < num_parameters; i++) { in ImportInstructionsImpl()

12