/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
D | dense_image_warp_test.py | 113 batch_index, argument 121 displacement = flows[batch_index, y_index, x_index, :] 137 top_left = image[batch_index, floor_y, floor_x, :] 138 top_right = image[batch_index, floor_y, ceil_x, :] 139 bottom_left = image[batch_index, ceil_y, floor_x, :] 140 bottom_right = image[batch_index, ceil_y, ceil_x, :] 152 pred_interpolation[batch_index, y_index, x_index, :], 205 batch_index = np.random.randint(0, shape[0]) 213 batch_index,
|
/external/tensorflow/tensorflow/contrib/layers/kernels/ |
D | sparse_feature_cross_kernel.cc | 148 void Update(const int64 batch_index, const int64 cross_count, in Update() argument 150 const int64 output_index = output_start_indices_[batch_index] + cross_count; in Update() 153 indices_matrix(output_index, 0) = batch_index; in Update() 175 string Generate(const int64 batch_index, in Generate() argument 181 cross_vec[i] = columns_[i]->Feature(batch_index, permutation[i]); in Generate() 200 int64 Generate(const int64 batch_index, in Generate() argument 207 int64 hash_i = columns_[i]->Feature(batch_index, permutation[i]); in Generate() 235 int64 Generate(const int64 batch_index, in Generate() argument 240 uint64 hash_i = columns_[i]->Feature(batch_index, permutation[i]); in Generate() 265 int64 batch_index) in ProductIterator() argument [all …]
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
D | mnist.py | 216 for (batch_index, images) in enumerate(dataset): 227 seed=batch_index) 258 if log_interval and batch_index > 0 and batch_index % log_interval == 0: 261 (batch_index, total_generator_loss / batch_index, 262 total_discriminator_loss / batch_index))
|
/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_cross_op.cc | 148 void Update(const int64 batch_index, const int64 cross_count, in Update() argument 150 const int64 output_index = output_start_indices_[batch_index] + cross_count; in Update() 153 indices_matrix(output_index, 0) = batch_index; in Update() 175 string Generate(const int64 batch_index, in Generate() argument 181 cross_vec[i] = columns_[i]->Feature(batch_index, permutation[i]); in Generate() 200 int64 Generate(const int64 batch_index, in Generate() argument 205 uint64 hash_i = columns_[i]->Feature(batch_index, permutation[i]); in Generate() 230 int64 batch_index) in ProductIterator() argument 231 : columns_(columns), batch_index_(batch_index) { in ProductIterator() 540 int batch_index) { in CrossCountByBatchIndex() argument [all …]
|
D | topk_op_gpu.h | 345 const int batch_index = blockIdx.x; 346 const T* batch_input = input + batch_index * length; 358 const int offset = batch_index * k;
|
D | batch_kernels.cc | 1103 auto batch_index = in OutputBatch() local 1107 auto available_it = available_tensors_.find(batch_index(i, 0)); in OutputBatch() 1155 const auto batch_index = in Compute() local 1158 const int64 batch_key = batch_index(i, 0); in Compute()
|
/external/tensorflow/tensorflow/core/util/ |
D | padding.cc | 55 const int32 batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding() local 57 if (explicit_paddings[2 * batch_index] != 0 || in CheckValidPadding() 58 explicit_paddings[2 * batch_index + 1] != 0 || in CheckValidPadding()
|
D | example_proto_helper.cc | 216 const Example& example, const string& example_name, const int batch_index, in SingleExampleProtoToTensors() argument 256 TF_RETURN_IF_ERROR(FeatureDenseCopy(batch_index, example_name, key, dtype, in SingleExampleProtoToTensors() 261 RowDenseCopy(batch_index, dtype, default_value, in SingleExampleProtoToTensors() 288 (*output_sparse_values_tmp)[d][batch_index] = in SingleExampleProtoToTensors() 289 FeatureSparseCopy(batch_index, key, dtype, f); in SingleExampleProtoToTensors() 291 (*output_sparse_values_tmp)[d][batch_index] = in SingleExampleProtoToTensors()
|
D | example_proto_helper.h | 77 const Example& example, const string& name, const int batch_index,
|
/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/python/keras/engine/ |
D | training_arrays.py | 326 for batch_index, (batch_start, batch_end) in enumerate(batches): 347 batch_logs = {'batch': batch_index, 'size': len(batch_ids)} 348 callbacks._call_batch_hook(mode, 'begin', batch_index, batch_logs) 349 progbar.on_batch_begin(batch_index, batch_logs) 357 if batch_index == 0: 363 callbacks._call_batch_hook(mode, 'end', batch_index, batch_logs) 364 progbar.on_batch_end(batch_index, batch_logs)
|
D | training_generator_test.py | 49 batch_index = i * batch_size % num_samples 51 start = batch_index
|
/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/python/keras/ |
D | metrics_correctness_test.py | 66 batch_index = i * batch_size % num_samples 68 start = batch_index 227 batch_index = i * batch_size % num_samples 229 start = batch_index
|
/external/tensorflow/tensorflow/contrib/batching/python/ops/ |
D | batch_ops.py | 176 batched_tensors, batch_index, id_t = gen_batch_ops.batch( 192 gen_batch_ops.unbatch(t, batch_index, id_t,
|
/external/mesa3d/src/mesa/state_tracker/ |
D | st_cb_perfmon.h | 34 unsigned batch_index; member
|
D | st_cb_perfmon.c | 100 cntr->batch_index = num_batch_counters; in init_perf_monitor() 304 result.batch[0] = stm->batch_result->batch[cntr->batch_index]; in st_GetPerfMonitorResult()
|
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
D | keras_support.py | 1791 batch_index = 0 1793 for batch_index, (batch_start, batch_end) in enumerate(batches): 1807 next_batch_logs['batch'] = batch_index 1809 if batch_index > 0: 1811 callbacks.on_batch_begin(batch_index - 1, last_batch_logs) 1819 if batch_index == 0: 1826 callbacks.on_batch_end(batch_index - 1, last_batch_logs) 1832 callbacks.on_batch_begin(batch_index, last_batch_logs) 1838 callbacks.on_batch_end(batch_index, last_batch_logs)
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | dot_op_emitter.cc | 911 llvm::Value* batch_index, in SliceOutInnerArray() argument 918 multidim_index[0] = batch_index; in SliceOutInnerArray() 920 batch_index->getType()); in SliceOutInnerArray()
|
/external/v8/src/arm64/ |
D | macro-assembler-arm64.cc | 1127 int batch_index = 0; in PushQueued() local 1129 batch[batch_index++] = queued_[index++]; in PushQueued() 1130 } while ((batch_index < 4) && (index < count) && in PushQueued() 1133 masm_->PushHelper(batch_index, batch[0].SizeInBytes(), in PushQueued() 1151 int batch_index = 0; in PopQueued() local 1153 batch[batch_index++] = queued_[index++]; in PopQueued() 1154 } while ((batch_index < 4) && (index < count) && in PopQueued() 1157 masm_->PopHelper(batch_index, batch[0].SizeInBytes(), in PopQueued()
|
/external/tensorflow/tensorflow/python/ops/ |
D | data_flow_ops.py | 2399 batch_index = index % self._batches 2400 batch_list[batch_index].append(protobuf)
|
/external/tensorflow/tensorflow/python/keras/layers/ |
D | recurrent.py | 514 batch_index, time_step_index = (1, 0) if self.time_major else (0, 1) 516 input_spec_shape[batch_index] = None
|