| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_cholesky_op.cc | 113 for (int64 batch_index = batch_begin; batch_index < batch_end; in Compute() local
114 ++batch_index) { in Compute()
118 num_rows, num_rows, input_matrix->nnz(batch_index), in Compute()
119 input_matrix->row_pointers_vec(batch_index).data(), in Compute()
120 input_matrix->col_indices_vec(batch_index).data(), in Compute()
121 input_matrix->values_vec<T>(batch_index).data()); in Compute()
133 permutation(permutation_indices_flat + batch_index * num_rows, in Compute()
145 invalid_input_index = batch_index; in Compute()
152 sparse_cholesky_factors[batch_index] = in Compute()
156 batch_ptr_vec(batch_index + 1) = in Compute()
[all …]
|
| D | sparse_ordering_amd_op.cc | 100 for (int64 batch_index = batch_begin; batch_index < batch_end; in Compute() local
101 ++batch_index) { in Compute()
107 num_rows, num_rows, input_matrix->nnz(batch_index), in Compute()
108 input_matrix->row_pointers_vec(batch_index).data(), in Compute()
109 input_matrix->col_indices_vec(batch_index).data(), in Compute()
110 input_matrix->col_indices_vec(batch_index).data()); in Compute()
120 permutation_indices.flat<int>().data() + batch_index * num_rows, in Compute()
|
| D | sparse_mat_mul_op.cc | 266 const CSRSparseMatrix& csr_matrix, const int rank, const int batch_index, in GetSparseMatrixRef() argument
273 num_rows, num_cols, csr_matrix.nnz(batch_index), in GetSparseMatrixRef()
274 csr_matrix.row_pointers_vec(batch_index).data(), in GetSparseMatrixRef()
275 csr_matrix.col_indices_vec(batch_index).data(), in GetSparseMatrixRef()
276 csr_matrix.values_vec<T>(batch_index).data()); in GetSparseMatrixRef()
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 74 size_t batch_index, in xnn_compute_spmm() argument
81 …(const void*) ((uintptr_t) context->input + batch_index * context->batched_input_stride + mr_block… in xnn_compute_spmm()
85 …(void*) ((uintptr_t) context->output + batch_index * context->batched_output_stride + mr_block_sta… in xnn_compute_spmm()
92 size_t batch_index, in xnn_compute_grouped_batch_igemm() argument
109 …(void*) ((uintptr_t) context->c + group_index * context->gc_stride + batch_index * context->bc_str… in xnn_compute_grouped_batch_igemm()
112 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_batch_igemm()
145 size_t batch_index, in xnn_compute_batch_igemm() argument
161 …(void*) ((uintptr_t) context->c + batch_index * context->bc_stride + mr_block_start * cm_stride + … in xnn_compute_batch_igemm()
164 context->a_offset + batch_index * context->ba_stride, in xnn_compute_batch_igemm()
196 size_t batch_index, in xnn_compute_grouped_subgemm2d() argument
[all …]
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 156 size_t batch_index,
199 size_t batch_index,
215 size_t batch_index,
234 size_t batch_index,
252 size_t batch_index,
287 size_t batch_index,
298 size_t batch_index,
333 size_t batch_index,
344 size_t batch_index,
377 size_t batch_index,
[all …]
|
| /external/XNNPACK/test/ |
| D | resize-bilinear-operator-tester.h | 241 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxF32() local
253 …output_ref[((batch_index * output_height() + output_y) * output_width() + output_x) * channels() +… in TestNHWCxF32()
254 …input[((batch_index * input_height() + input_y_top) * input_width() + input_x_left) * input_pixel_… in TestNHWCxF32()
255 …input[((batch_index * input_height() + input_y_top) * input_width() + input_x_right) * input_pixel… in TestNHWCxF32()
256 …input[((batch_index * input_height() + input_y_bottom) * input_width() + input_x_left) * input_pix… in TestNHWCxF32()
257 …input[((batch_index * input_height() + input_y_bottom) * input_width() + input_x_right) * input_pi… in TestNHWCxF32()
326 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNCHWxF32() local
338 …output_ref[batch_index * output_num_elements + c * output_num_pixels + output_y * output_width() +… in TestNCHWxF32()
339 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_top * input_width() + inpu… in TestNCHWxF32()
340 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_top * input_width() + inpu… in TestNCHWxF32()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 52 const int32 batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding() local
54 if (explicit_paddings[2 * batch_index] != 0 || in CheckValidPadding()
55 explicit_paddings[2 * batch_index + 1] != 0 || in CheckValidPadding()
|
| D | example_proto_helper.cc | 219 const Example& example, const string& example_name, const int batch_index, in SingleExampleProtoToTensors() argument
259 TF_RETURN_IF_ERROR(FeatureDenseCopy(batch_index, example_name, key, dtype, in SingleExampleProtoToTensors()
264 RowDenseCopy(batch_index, dtype, default_value, in SingleExampleProtoToTensors()
291 (*output_sparse_values_tmp)[d][batch_index] = in SingleExampleProtoToTensors()
292 FeatureSparseCopy(batch_index, key, dtype, f); in SingleExampleProtoToTensors()
294 (*output_sparse_values_tmp)[d][batch_index] = in SingleExampleProtoToTensors()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_UnbatchGrad.pbtxt | 5 Acts like Batch but using the given batch_index index of batching things as they
10 batch_index: The batch_index given to the Unbatch operation this is the gradient
|
| D | api_def_Unbatch.pbtxt | 14 batch_index: The matching batch_index obtained from Batch.
|
| D | api_def_Batch.pbtxt | 15 Each op which emits a non-empty batch will also emit a non-empty batch_index
34 batch_index: If out_tensors is non-empty, has information to invert it.
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | reader_dataset_ops_test_base.py | 212 batch_index = 0
228 [[batch_index, i] for i in range(len(keywords))])
229 batch_index += 1
241 batch_index = 0
349 batch_index = 0
360 batch_index += 1
364 batch_index = 0
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_cross_op.cc | 269 void Update(const int64 batch_index, const int64 cross_count, in Update() argument
271 const int64 output_index = output_start_indices_[batch_index] + cross_count; in Update()
274 indices_matrix(output_index, 0) = batch_index; in Update()
297 string Generate(const int64 batch_index, const std::vector<int>& permutation, in Generate() argument
301 cross_vec[i] = columns_[i]->Feature(batch_index, permutation[i], false); in Generate()
322 int64 Generate(const int64 batch_index, const std::vector<int>& permutation, in Generate() argument
327 uint64 hash_i = columns_[i]->Feature(batch_index, permutation[i], false); in Generate()
354 int64 Generate(const int64 batch_index, const std::vector<int>& permutation, in Generate() argument
358 columns_[0]->Feature(batch_index, permutation[0], strong_hash); in Generate()
361 columns_[i]->Feature(batch_index, permutation[i], strong_hash); in Generate()
[all …]
|
| D | ragged_cross_op.cc | 204 void WriteCombination(int64 batch_index, const std::vector<int>& combination, in WriteCombination() argument
209 features_[i]->ReadValue(batch_index, combination[i], &cross_vec[i]); in WriteCombination()
216 void WriteCombination(int64 batch_index, const std::vector<int>& combination, in WriteCombination() argument
222 features_[i]->ReadValue(batch_index, combination[i], &hash_i); in WriteCombination()
235 void NextCombination(int64 batch_index, std::vector<int>* combination) const { in NextCombination() argument
241 if ((*combination)[i] == features_[i]->FeatureCount(batch_index)) { in NextCombination()
580 int batch_index) { in CrossCountByBatchIndex() argument
583 const auto feature_count = features[i]->FeatureCount(batch_index); in CrossCountByBatchIndex()
|
| D | in_topk_op_gpu.cu.cc | 45 const int batch_index = i / num_classes; in ComputePredictionMaskKernel() local
46 TargetT target_idx = ldg(targets + batch_index); in ComputePredictionMaskKernel()
55 ldg(predictions + batch_index * num_classes + target_idx); in ComputePredictionMaskKernel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | topk_op_test.py | 85 for batch_index in range(shape[0]):
87 if np.isclose(values[batch_index, index],
88 values[batch_index, index + 1]):
89 self.assertLess(indices[batch_index, index],
90 indices[batch_index, index + 1])
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Batch.pbtxt | 12 name: "batch_index"
82 name: "batch_index"
|
| D | Unbatch.pbtxt | 8 name: "batch_index"
|
| D | UnbatchGrad.pbtxt | 8 name: "batch_index"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Batch.pbtxt | 12 name: "batch_index"
82 name: "batch_index"
|
| D | Unbatch.pbtxt | 8 name: "batch_index"
|
| D | UnbatchGrad.pbtxt | 8 name: "batch_index"
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays_v1.py | 353 for batch_index, (batch_start, batch_end) in enumerate(batches):
380 batch_logs = {'batch': batch_index, 'size': len(batch_ids)}
381 callbacks._call_batch_hook(mode, 'begin', batch_index, batch_logs)
389 if batch_index == 0:
395 callbacks._call_batch_hook(mode, 'end', batch_index, batch_logs)
|
| D | training_generator_test.py | 52 batch_index = i * batch_size % num_samples
54 start = batch_index
78 batch_index = i * batch_size % num_samples
80 start = batch_index
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_cb_perfmon.h | 34 unsigned batch_index; member
|