Home
last modified time | relevance | path

Searched refs:batch_index (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/
Ddense_image_warp_test.py113 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/
Dsparse_feature_cross_kernel.cc148 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/
Dmnist.py216 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/
Dsparse_cross_op.cc148 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 …]
Dtopk_op_gpu.h345 const int batch_index = blockIdx.x;
346 const T* batch_input = input + batch_index * length;
358 const int offset = batch_index * k;
Dbatch_kernels.cc1103 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/
Dpadding.cc55 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()
Dexample_proto_helper.cc216 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()
Dexample_proto_helper.h77 const Example& example, const string& name, const int batch_index,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_UnbatchGrad.pbtxt5 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
Dapi_def_Unbatch.pbtxt14 batch_index: The matching batch_index obtained from Batch.
Dapi_def_Batch.pbtxt15 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/
Dreader_dataset_ops_test_base.py212 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/
Dtraining_arrays.py326 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)
Dtraining_generator_test.py49 batch_index = i * batch_size % num_samples
51 start = batch_index
/external/tensorflow/tensorflow/python/kernel_tests/
Dtopk_op_test.py85 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/
Dmetrics_correctness_test.py66 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/
Dbatch_ops.py176 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/
Dst_cb_perfmon.h34 unsigned batch_index; member
Dst_cb_perfmon.c100 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/
Dkeras_support.py1791 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/
Ddot_op_emitter.cc911 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/
Dmacro-assembler-arm64.cc1127 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/
Ddata_flow_ops.py2399 batch_index = index % self._batches
2400 batch_list[batch_index].append(protobuf)
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent.py514 batch_index, time_step_index = (1, 0) if self.time_major else (0, 1)
516 input_spec_shape[batch_index] = None

12