| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators.cc | 61 int64_t 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_t i = 0; i < parameter_count; i++) { in CreateScalarComparisonComputation()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | zip_test_utils.py | 381 parameter_count = 0
383 parameter_count += functools.reduce(
386 all_parameter_count = parameter_count
388 all_parameter_count += options.multi_gen_state.parameter_count
397 options.multi_gen_state.parameter_count = all_parameter_count
413 parameter_count = 0
419 parameter_count += functools.reduce(
426 parameter_count = 0
431 parameter_count += functools.reduce(operator.mul, [
697 tf_failures = parameter_count - tf_success
[all …]
|
| D | generate_examples_lib.py | 222 self.parameter_count = 0
|
| /external/pytorch/torch/onnx/_internal/exporter/ |
| D | _analysis.py | 27 parameter_count: defaultdict[torch.dtype, int] = dataclasses.field( variable in ModelInfo
56 parameter_count: defaultdict[torch.dtype, int] = defaultdict(int)
60 parameter_count[dtype] += parameter.numel()
66 return parameter_count, buffer_count
194 model_info.parameter_count, model_info.buffer_count = _count_weights(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_computation.cc | 61 int parameter_count = 0; in Build() local
64 parameter_count++; in Build()
72 name_, parameter_count, &instructions_, root, fusion_instruction_)); in Build()
76 const std::string& name, int parameter_count, in HloComputation() argument
86 param_instructions_.resize(parameter_count, nullptr); in HloComputation()
91 CHECK(param_no >= 0 && param_no < parameter_count) in HloComputation()
93 << parameter_count << "), got " << param_no; in HloComputation()
662 int64_t parameter_count = 0; in CreateFromProto() local
669 parameter_count++; in CreateFromProto()
688 std::vector<bool> parameters_seen(parameter_count); in CreateFromProto()
[all …]
|
| D | computation_layout.h | 59 int parameter_count() const { return parameter_layouts_.size(); } in parameter_count() function
|
| D | hlo_module.cc | 401 module_config.entry_computation_layout().parameter_count()); in CreateFromProto()
573 for (int64_t i = 0; i < entry_layout->parameter_count(); ++i) { in CreateModuleConfigFromShape()
633 int64_t parameter_count = 0; in OutlineExpressionFromComputation() local
653 parameter_count, old_operand->shape(), "p")); in OutlineExpressionFromComputation()
654 ++parameter_count; in OutlineExpressionFromComputation()
|
| D | hlo_module_util.cc | 167 for (int i = 0; i < module->entry_computation_layout().parameter_count(); in UpdateEntryComputationLayout()
|
| D | hlo_computation.h | 666 const std::string& name, int parameter_count,
|
| D | layout_assignment.cc | 674 called_computation_layout.parameter_count()); in AddMandatoryConstraints()
2367 for (int64_t i = 0; i < computed_computation_layout.parameter_count(); ++i) { in PropagateComputationLayouts()
2460 TF_RET_CHECK(entry_computation_layout_->parameter_count() == in Run()
|
| D | hlo_verifier.cc | 1826 if (computation->num_parameters() != layout.parameter_count()) { in VerifyEntryComputationLayout()
1830 layout.parameter_count(), computation->num_parameters()); in VerifyEntryComputationLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_conversions.cc | 446 for (int i = 0; i < se_config.entry_computation_layout.parameter_count; ++i) { in ConfigWithLayout()
486 new XLA_Shape[layout.parameter_count()]; in ToC()
487 for (int i = 0; i < layout.parameter_count(); ++i) { in ToC()
492 hlo_config.entry_computation_layout.parameter_count = in ToC()
493 layout.parameter_count(); in ToC()
531 for (auto i = 0; i < c_config->entry_computation_layout.parameter_count; in Destroy()
|
| D | c_api_decl.h | 298 int parameter_count; member
|
| /external/mesa3d/src/compiler/glsl/ |
| D | ast_function.cpp | 1259 const unsigned parameter_count = in process_vec_mat_constructor() local
1262 if (parameter_count == 0 in process_vec_mat_constructor()
1264 constructor_type->vector_elements != parameter_count) in process_vec_mat_constructor()
1266 constructor_type->matrix_columns != parameter_count)) { in process_vec_mat_constructor()
1364 const unsigned parameter_count = in process_array_constructor() local
1368 if ((parameter_count == 0) || in process_array_constructor()
1369 (!is_unsized_array && (constructor_type->length != parameter_count))) { in process_array_constructor()
1383 parameter_count, 0); in process_array_constructor()
1385 assert(constructor_type->length == parameter_count); in process_array_constructor()
1435 glsl_array_type(element_type, parameter_count, 0); in process_array_constructor()
[all …]
|
| /external/google-breakpad/src/client/mac/crash_generation/ |
| D | Inspector.h | 57 unsigned int parameter_count; // key-value pairs member
|
| D | Inspector.mm | 222 printf("parameter count = %d\n", info.parameter_count);
234 for (unsigned int i = 0; i < info.parameter_count; ++i) {
256 if (parameters_read != info.parameter_count) {
|
| /external/mesa3d/src/gallium/frontends/vdpau/ |
| D | mixer.c | 44 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/ |
| D | xrt_execute_op.cc | 80 input_is_dynamic.reserve(computation_layout.parameter_count()); in GetDynamicInputInfo()
81 for (int64_t 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/ |
| D | local_client.cc | 127 if (argument_shapes_size != computation_layout.parameter_count()) { in RunHelper()
130 computation_layout.parameter_count(), argument_shapes.size()); in RunHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | params_test.cc | 222 constexpr int parameter_count = 100; in XLA_TEST_F() local
223 for (int i = 0; i < parameter_count; ++i) { in XLA_TEST_F()
|
| D | hlo_test_base.h | 303 module->mutable_entry_computation_layout()->parameter_count()); in ForceParameterLayout()
|
| /external/rust/android-crates-io/crates/rusqlite/src/ |
| D | statement.rs | 473 let count = self.parameter_count(); in ensure_parameter_count()
499 pub fn parameter_count(&self) -> usize { in parameter_count() method
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tfrt_cpu_pjrt_client.cc | 1187 if (computation_layout.parameter_count() == 0) { in TfrtCpuExecutable()
1192 if (computation_layout.parameter_count() > 1 || in TfrtCpuExecutable()
1194 input_buffer_sizes_in_bytes_.reserve(computation_layout.parameter_count()); in TfrtCpuExecutable()
1195 for (int i = 0; i < computation_layout.parameter_count(); ++i) { in TfrtCpuExecutable()
|
| D | pjrt_stream_executor_client.cc | 1781 parameter_shapes.reserve(computation_layout.parameter_count()); in PjRtStreamExecutorExecutable()
1782 for (int i = 0; i < computation_layout.parameter_count(); ++i) { in PjRtStreamExecutorExecutable()
|
| /external/swiftshader/third_party/subzero/src/ |
| D | WasmTranslator.cpp | 1063 const auto NumArgs = Sig->parameter_count(); in CallDirect()
1096 const auto NumArgs = Sig->parameter_count(); in CallImport()
|