Home
last modified time | relevance | path

Searched refs:parameter_count (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc61 int64 parameter_count = 0; in CreateScalarComparisonComputation() local
73 auto lhs_param = Parameter(b.get(), parameter_count * 2, scalar_shape, in CreateScalarComparisonComputation()
74 absl::StrCat("p.", parameter_count, ".lhs")); in CreateScalarComparisonComputation()
75 auto rhs_param = Parameter(b.get(), parameter_count * 2 + 1, scalar_shape, in CreateScalarComparisonComputation()
76 absl::StrCat("p.", parameter_count, ".rhs")); in CreateScalarComparisonComputation()
79 if (generators[parameter_count].has_value()) { in CreateScalarComparisonComputation()
80 last_generator_index = parameter_count; in CreateScalarComparisonComputation()
82 parameter_count++; in CreateScalarComparisonComputation()
85 CHECK_NE(parameter_count, 0); in CreateScalarComparisonComputation()
98 for (int64 i = 0; i < parameter_count; i++) { in CreateScalarComparisonComputation()
/external/tensorflow/tensorflow/lite/testing/
Dzip_test_utils.py301 parameter_count = 0
303 parameter_count += functools.reduce(
306 all_parameter_count = parameter_count
308 all_parameter_count += options.multi_gen_state.parameter_count
317 options.multi_gen_state.parameter_count = all_parameter_count
334 parameter_count = 0
339 parameter_count += functools.reduce(operator.mul, [
556 tf_failures = parameter_count - tf_success
558 if tf_failures / parameter_count > 0.8:
561 (zip_path, int(100 * tf_failures / parameter_count)))
Dgenerate_examples_lib.py199 self.parameter_count = 0
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_computation.cc53 int parameter_count = 0; in Build() local
56 parameter_count++; in Build()
64 name_, parameter_count, &instructions_, root, fusion_instruction_)); in Build()
68 const string& name, int parameter_count, in HloComputation() argument
75 param_instructions_.resize(parameter_count, nullptr); in HloComputation()
80 CHECK(param_no >= 0 && param_no < parameter_count) in HloComputation()
82 << parameter_count << "), got " << param_no; in HloComputation()
669 int64 parameter_count = 0; in CreateFromProto() local
676 parameter_count++; in CreateFromProto()
695 std::vector<bool> parameters_seen(parameter_count); in CreateFromProto()
[all …]
Dhlo_module.cc342 module_config.entry_computation_layout().parameter_count()); in CreateFromProto()
472 for (int64 i = 0; i < entry_layout->parameter_count(); ++i) { in CreateModuleConfigFromShape()
519 int64 parameter_count = 0; in OutlineExpressionFromComputation() local
539 parameter_count, old_operand->shape(), "p")); in OutlineExpressionFromComputation()
540 ++parameter_count; in OutlineExpressionFromComputation()
Dcomputation_layout.h59 int parameter_count() const { return parameter_layouts_.size(); } in parameter_count() function
Dhlo_computation.h497 const string& name, int parameter_count,
Dlayout_assignment.cc559 called_computation_layout.parameter_count()); in AddMandatoryConstraints()
2051 for (int64 i = 0; i < computed_computation_layout.parameter_count(); ++i) { in PropagateComputationLayouts()
2144 TF_RET_CHECK(entry_computation_layout_->parameter_count() == in Run()
Dhlo_verifier.cc1292 if (computation->num_parameters() != layout.parameter_count()) { in VerifyEntryComputationLayout()
1296 layout.parameter_count(), computation->num_parameters()); in VerifyEntryComputationLayout()
Dhlo_parser.cc5195 int64 parameter_count = 0; in ParseSingleInstruction() local
5197 [this, &builder, &parameter_count]( in ParseSingleInstruction()
5200 std::string new_name = name.empty() ? StrCat("_", parameter_count) : name; in ParseSingleInstruction()
5202 HloInstruction::CreateParameter(parameter_count++, shape, new_name)); in ParseSingleInstruction()
Dservice.cc848 executable->module_config().entry_computation_layout().parameter_count(); in Execute()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc406 for (int i = 0; i < se_config.entry_computation_layout.parameter_count; ++i) { in ConfigWithLayout()
441 new XLA_Shape[layout.parameter_count()]; in ToC()
442 for (int i = 0; i < layout.parameter_count(); ++i) { in ToC()
447 hlo_config.entry_computation_layout.parameter_count = in ToC()
448 layout.parameter_count(); in ToC()
476 for (auto i = 0; i < c_config->entry_computation_layout.parameter_count; in Free()
Dc_api_decl.h271 int parameter_count; member
/external/mesa3d/src/compiler/glsl/
Dast_function.cpp1202 const unsigned parameter_count = in process_vec_mat_constructor() local
1205 if (parameter_count == 0 in process_vec_mat_constructor()
1207 constructor_type->vector_elements != parameter_count) in process_vec_mat_constructor()
1209 constructor_type->matrix_columns != parameter_count)) { in process_vec_mat_constructor()
1308 const unsigned parameter_count = in process_array_constructor() local
1312 if ((parameter_count == 0) || in process_array_constructor()
1313 (!is_unsized_array && (constructor_type->length != parameter_count))) { in process_array_constructor()
1327 parameter_count); in process_array_constructor()
1329 assert(constructor_type->length == parameter_count); in process_array_constructor()
1380 parameter_count); in process_array_constructor()
[all …]
/external/google-breakpad/src/client/mac/crash_generation/
DInspector.h58 unsigned int parameter_count; // key-value pairs member
DInspector.mm223 printf("parameter count = %d\n", info.parameter_count);
235 for (unsigned int i = 0; i < info.parameter_count; ++i) {
257 if (parameters_read != info.parameter_count) {
/external/mesa3d/src/gallium/frontends/vdpau/
Dmixer.c44 uint32_t parameter_count, in vlVdpVideoMixerCreate() argument
127 for (i = 0; i < parameter_count; ++i) { in vlVdpVideoMixerCreate()
906 uint32_t parameter_count, in vlVdpVideoMixerGetParameterValues() argument
915 if (!parameter_count) in vlVdpVideoMixerGetParameterValues()
919 for (i = 0; i < parameter_count; ++i) { in vlVdpVideoMixerGetParameterValues()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc80 input_is_dynamic.reserve(computation_layout.parameter_count()); in GetDynamicInputInfo()
81 for (int64 i = 0; i < computation_layout.parameter_count(); ++i) { in GetDynamicInputInfo()
96 input_coords, working_set, backend, computation_layout.parameter_count(), in GetInputTuples()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc126 if (argument_shapes_size != computation_layout.parameter_count()) { in RunHelper()
129 computation_layout.parameter_count(), argument_shapes.size()); in RunHelper()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dparams_test.cc224 constexpr int parameter_count = 100; in XLA_TEST_F() local
225 for (int i = 0; i < parameter_count; ++i) { in XLA_TEST_F()
Dhlo_test_base.h268 module->mutable_entry_computation_layout()->parameter_count()); in ForceParameterLayout()
/external/rust/crates/rusqlite/src/
Dstatement.rs475 pub fn parameter_count(&self) -> usize { in parameter_count() method
/external/swiftshader/third_party/subzero/src/
DWasmTranslator.cpp1063 const auto NumArgs = Sig->parameter_count(); in CallDirect()
1096 const auto NumArgs = Sig->parameter_count(); in CallImport()
/external/google-breakpad/src/client/mac/Framework/
DBreakpad.mm716 info.parameter_count = config_params_->GetCount();