| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | dataset_serialization_test_base.py | 64 def run_core_tests(self, ds_fn1, ds_fn2, num_outputs, sparse_tensors=False): argument
86 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
88 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
90 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
92 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
94 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
96 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
98 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
107 num_outputs,
112 num_outputs, argument
[all …]
|
| D | stats_dataset_serialization_test.py | 52 num_outputs = 100
54 lambda: self._build_dataset_bytes_stats(num_outputs),
55 lambda: self._build_dataset_bytes_stats(num_outputs // 10), num_outputs)
79 num_outputs = 100
82 lambda: self._build_dataset_latency_stats(num_outputs),
83 lambda: self._build_dataset_latency_stats(num_outputs // 10),
84 num_outputs)
86 self.run_core_tests(lambda: self._build_dataset_multiple_tags(num_outputs),
87 None, num_outputs)
92 lambda: self._build_dataset_multiple_tags(num_outputs, tag1, tag2),
[all …]
|
| D | map_dataset_serialization_test.py | 93 num_outputs = 100
101 return dataset_ops.Dataset.range(num_outputs).map(defun_fn)
103 self.run_core_tests(_build_ds, None, num_outputs)
106 num_outputs = 100
120 return dataset_ops.Dataset.range(num_outputs).map(defun_fn)
122 self.run_core_tests(_build_ds, None, num_outputs)
132 def _build_ds(num_outputs): argument
133 return dataset_ops.Dataset.range(num_outputs).map(_sparse)
135 num_outputs = 10
136 self.run_core_tests(lambda: _build_ds(num_outputs),
[all …]
|
| D | shuffle_dataset_serialization_test.py | 49 num_outputs = range_limit * num_repeats
68 num_outputs)
76 num_outputs = range_limit * num_repeats
97 num_outputs=num_outputs,
103 break_points=self.gen_break_points(num_outputs),
104 num_outputs=num_outputs,
112 num_outputs = range_limit * num_repeats
141 expected = [self.evaluate(get_next_ops) for _ in range(num_outputs)]
143 actual = [self.evaluate(get_next_ops) for _ in range(num_outputs)]
|
| D | serialization_integration_test.py | 31 def _build_input_pipeline(self, name, num_outputs): argument
33 ds = dataset_ops.Dataset.range(num_outputs).shuffle(
40 def _build_graph(self, num_pipelines, num_outputs): argument
45 init_op, get_next_op = self._build_input_pipeline(name, num_outputs)
56 num_outputs = 100
61 num_outputs)
72 num_outputs)
75 for _ in range(num_outputs - break_point):
81 self.assertSequenceEqual(sorted(output), range(num_outputs))
|
| D | tf_record_dataset_serialization_test.py | 68 num_outputs = num_epochs * self._num_files * self._num_records // batch_size
74 num_outputs)
77 num_outputs * batch_size)
82 num_outputs = num_epochs * self._num_files * self._num_records
85 num_outputs)
89 num_outputs = num_epochs * self._num_files * self._num_records
92 lambda: self._build_iterator_graph(num_epochs * 2), num_outputs)
95 lambda: self._build_iterator_graph(num_epochs * 2), num_outputs)
|
| D | cache_dataset_serialization_test.py | 37 self.num_outputs = self.range_size * self.num_repeats
71 self.num_outputs - 5,
93 self.num_outputs - 5,
103 self.num_outputs - 5,
123 self.num_outputs - 15,
142 self.num_outputs - 15,
166 self.num_outputs - 5,
183 ds_fn, [], self.num_outputs, ckpt_saved=True, verify_exhausted=False)
206 self.num_outputs - 5,
224 ds_fn, [], self.num_outputs, verify_exhausted=False)
[all …]
|
| D | filter_dataset_serialization_test.py | 36 num_outputs = sum(x % 3 != 2 for x in range(100))
39 num_outputs)
48 num_outputs = sum((x**2) % 2 == 0 for x in range(10))
49 self.run_core_tests(self._build_filter_dict_graph, None, num_outputs)
64 num_outputs = 5
65 self.run_core_tests(self._build_sparse_filter, None, num_outputs)
|
| D | parallel_interleave_dataset_serialization_test.py | 36 self.num_outputs = np.sum(self.input_values) * 2
52 self.num_outputs)
57 None, self.num_outputs)
62 None, self.num_outputs)
65 break_points = self.gen_break_points(self.num_outputs, 10)
73 break_points, self.num_outputs)
|
| D | parallel_map_dataset_serialization_test.py | 105 num_outputs = 100
113 return dataset_ops.Dataset.range(num_outputs).map(
116 self.run_core_tests(_build_ds, None, num_outputs)
119 num_outputs = 100
133 return dataset_ops.Dataset.range(num_outputs).map(
136 self.run_core_tests(_build_ds, None, num_outputs)
|
| D | batch_dataset_serialization_test.py | 44 num_outputs = tensor_slice_len // batch_size
48 num_outputs)
59 num_outputs = len(components) // 4
62 num_outputs)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 44 def execute(op_name, num_outputs, inputs, attrs=None): argument
46 op_name, num_outputs, inputs, attrs, context.context())
298 num_outputs=1,
309 num_outputs=1,
318 num_outputs=1,
330 num_outputs=50,
341 num_outputs=0,
353 num_outputs=1,
362 num_outputs=1,
372 num_outputs=1,
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | split_benchmark.py | 58 def _run_graph(self, device, output_shape, variable, num_outputs, axis): argument
75 input_shape = [output_shape[0] * num_outputs, output_shape[1]]
76 sizes = [output_shape[0] for _ in range(num_outputs)]
78 input_shape = [output_shape[0], output_shape[1] * num_outputs]
79 sizes = [output_shape[1] for _ in range(num_outputs)]
84 size=num_outputs)
114 num_outputs = 100
119 self._run_graph("gpu", shape, v, num_outputs, axis)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_ops.cc | 31 if (c->num_inputs() < c->num_outputs()) { in __anon8af73ba60102()
39 for (int i = 0; i < c->num_outputs(); ++i) { in __anon8af73ba60102()
121 if (output_shapes.size() != c->num_outputs()) { in __anon8af73ba60202()
124 output_shapes.size(), " vs. ", c->num_outputs()); in __anon8af73ba60202()
150 if (output_shapes.size() != c->num_outputs()) { in __anon8af73ba60302()
153 output_shapes.size(), " vs. ", c->num_outputs()); in __anon8af73ba60302()
173 for (int i = 0; i < c->num_outputs(); ++i) { in __anon8af73ba60402()
210 if (output_shapes.size() != c->num_outputs()) { in __anon8af73ba60502()
213 output_shapes.size(), " vs. ", c->num_outputs()); in __anon8af73ba60502()
222 for (int i = 0; i < c->num_outputs(); ++i) { in __anon8af73ba60502()
[all …]
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | ml_sse3.c | 166 const int num_outputs = output_layer ? nn_config->num_outputs in av1_nn_predict_sse3() local
169 if (num_inputs % 4 == 0 && num_outputs % 8 == 0) { in av1_nn_predict_sse3()
170 for (int out = 0; out < num_outputs; out += 8) { in av1_nn_predict_sse3()
182 } else if (num_inputs % 8 == 0 && num_outputs % 4 == 0) { in av1_nn_predict_sse3()
183 for (int out = 0; out < num_outputs; out += 4) { in av1_nn_predict_sse3()
193 } else if (num_inputs % 4 == 0 && num_outputs % 4 == 0) { in av1_nn_predict_sse3()
194 for (int out = 0; out < num_outputs; out += 4) { in av1_nn_predict_sse3()
205 for (int out = 0; out < num_outputs; out++) { in av1_nn_predict_sse3()
215 for (int out = 0; out < num_outputs; out++) { in av1_nn_predict_sse3()
227 for (int out = 0; out < num_outputs; out++) { in av1_nn_predict_sse3()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 125 void CostModel::Ensure(int id, int num_outputs) { in Ensure() argument
134 if (num_outputs > 0) { in Ensure()
139 CHECK_LE(perslot->size(), num_outputs); in Ensure()
145 perslot->resize(num_outputs, Bytes(-1)); in Ensure()
146 output_port_alloc_ids->resize(num_outputs, -1); in Ensure()
147 max_mem_usage->output_port_mem.resize(num_outputs, Bytes(-1)); in Ensure()
148 max_mem_usage->output_port_shape.resize(num_outputs, unknown_shape_); in Ensure()
149 max_mem_usage->output_port_type.resize(num_outputs, DT_INVALID); in Ensure()
153 void CostModel::SetNumOutputs(const Node* node, int num_outputs) { in SetNumOutputs() argument
161 CHECK_EQ(num_outputs, perslot->size()) in SetNumOutputs()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 95 kernel_state.num_outputs = n->num_outputs(); in Initialize()
112 kernel_state.output_locations.resize(kernel_state.num_outputs); in Initialize()
123 kernel_state.output_alloc_attrs.resize(kernel_state.num_outputs); in Initialize()
127 for (int out = 0; out < n->num_outputs(); out++) { in Initialize()
245 const size_t num_outputs = kernel_state.num_outputs; in RunAsync() local
260 OpKernelContext ctx(¶ms, num_outputs); in RunAsync()
273 for (size_t k = 0; k < executed_kernel_state.num_outputs; ++k) { in RunAsync()
294 for (size_t j = 0; j < num_outputs; ++j) { in RunAsync()
332 size_t num_outputs; member
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.cc | 39 num_outputs_(params.num_outputs()) {} in GrowStats()
273 const float num_classes = params_.num_outputs(); in CheckPruneHoeffding()
299 float range = 0.25 * static_cast<float>(params_.num_outputs()) * weight_sum_; in CheckFinishEarlyHoeffding()
420 const int32 num_classes = params_.num_outputs(); in ExtractFromProto()
493 for (int i = 0; i < params_.num_outputs(); ++i) { in InitLeafClassStats()
499 for (int i = 0; i < params_.num_outputs(); ++i) { in InitLeafClassStats()
584 const int32 num_classes = params_.num_outputs(); in GiniScore()
740 params_.num_outputs()); in ExtractFromProto()
745 params_.num_outputs()); in ExtractFromProto()
775 const int32 num_classes = params_.num_outputs(); in GiniScore()
[all …]
|
| D | leaf_model_operators.cc | 57 QCHECK_LT(int_label, params_.num_outputs()) in UpdateModel()
68 for (int i = 0; i < params_.num_outputs(); ++i) { in InitModel()
92 QCHECK_LT(int_label, params_.num_outputs()) in UpdateModel()
147 for (int i = 0; i < params_.num_outputs(); ++i) { in InitModel()
155 for (int i = 0; i < params_.num_outputs(); ++i) { in ExportModel()
|
| /external/mesa3d/src/gallium/drivers/svga/ |
| D | svga_pipe_streamout.c | 59 assert(info->num_outputs <= PIPE_MAX_SO_OUTPUTS); in svga_create_stream_output()
67 assert(info->num_outputs <= SVGA3D_MAX_STREAMOUT_DECLS); in svga_create_stream_output()
86 info->num_outputs, id); in svga_create_stream_output()
92 for (i = 0; i < info->num_outputs; i++) { in svga_create_stream_output()
105 decls[i].registerIndex = shader->info.num_outputs; in svga_create_stream_output()
119 decls[i].registerIndex = shader->info.num_outputs + 1 + in svga_create_stream_output()
139 info->num_outputs, in svga_create_stream_output()
145 info->num_outputs, in svga_create_stream_output()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | operation_jni.cc | 95 int num_outputs = TF_OperationNumOutputs(op); in Java_org_tensorflow_Operation_shape() local
96 if (output_index < 0 || output_index >= num_outputs) { in Java_org_tensorflow_Operation_shape()
100 output_index, num_outputs); in Java_org_tensorflow_Operation_shape()
148 int num_outputs = TF_OperationNumOutputs(op); in Java_org_tensorflow_Operation_dtype() local
149 if (output_index < 0 || output_index >= num_outputs) { in Java_org_tensorflow_Operation_dtype()
153 output_index, num_outputs); in Java_org_tensorflow_Operation_dtype()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | linalg_ops_common.cc | 165 const int num_outputs = output_matrix_shapes->size(); in PrepareOutputs() local
169 context, num_outputs <= context->num_outputs(), in PrepareOutputs()
172 num_outputs, context->num_outputs())); in PrepareOutputs()
179 for (int output_idx = 0; output_idx < context->num_outputs(); ++output_idx) { in PrepareOutputs()
181 if (output_idx < num_outputs) { in PrepareOutputs()
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone_test.cc | 171 for (int num_outputs = 0; !end_of_input; ++num_outputs) { in TEST() local
177 test_case.expected_outputs[num_outputs]); in TEST()
179 EXPECT_EQ(test_case.expected_outputs.size(), num_outputs); in TEST()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column_ops.py | 273 num_outputs, argument
294 shape=[embedding_lookup_arguments.vocab_size, num_outputs],
314 num_outputs, argument
342 shape=[prev_size, num_outputs],
362 num_outputs, argument
416 num_outputs,
421 shape=[num_outputs],
433 num_outputs, argument
503 num_outputs,
517 shape=[tensor.get_shape()[1], num_outputs],
[all …]
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | layers.py | 97 num_outputs, argument
220 filters=num_outputs,
261 num_outputs, argument
317 if not isinstance(num_outputs, six.integer_types):
319 (num_outputs,))
333 units=num_outputs,
|