| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 67 def fully_connected_model_fn(batch_size, activation_size, num_layers): argument
73 def lstm_model_fn(batch_size, state_size, steps, inputs_size=None): argument
86 def dynamic_lstm_model_fn(batch_size, state_size, max_steps): argument
104 def create_fc_batch_jacobian(batch_size, activation_size, num_layers): argument
112 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
120 def create_dynamic_lstm_batch_jacobian(batch_size, state_size, max_steps): argument
134 def create_lstm_batch_hessian(batch_size, state_size, steps): argument
146 def create_lstm_hessian(batch_size, state_size, steps): argument
162 def create_fc_per_eg_grad(batch_size, activation_size, num_layers): argument
186 def create_lstm_per_eg_grad(batch_size, state_size, steps, inputs_size=None): argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | batching.py | 29 def dense_to_sparse_batch(batch_size, row_shape): argument
101 def batch_and_drop_remainder(batch_size): argument
140 def padded_batch_and_drop_remainder(batch_size, argument
233 batch_size, argument
|
| /external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
| D | cifar10_input.py | 102 batch_size, shuffle): argument
140 def distorted_inputs(data_dir, batch_size): argument
206 def inputs(eval_data, data_dir, batch_size): argument
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/ |
| D | densenet_graph_test.py | 32 def image_shape(batch_size): argument
38 def random_batch(batch_size): argument
84 def _report(self, label, start, num_iters, batch_size): argument
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/ |
| D | resnet50_graph_test.py | 35 def image_shape(batch_size): argument
41 def random_batch(batch_size): argument
109 def _report(self, label, start, num_iters, batch_size): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ctc_ops.cc | 54 DimensionHandle batch_size; in __anon78c05cd60102() local
80 DimensionHandle batch_size; in __anon78c05cd60202() local
110 DimensionHandle batch_size; in __anon78c05cd60302() local
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | graph_io.py | 50 batch_size, argument
118 batch_size, argument
188 batch_size, argument
339 batch_size, argument
466 batch_size, argument
555 batch_size, argument
759 batch_size, argument
839 batch_size, argument
|
| D | data_feeder.py | 48 def _get_in_out_shape(x_shape, y_shape, n_classes, batch_size=None): argument
114 batch_size=None, argument
162 def _batch_data(x, batch_size=None): argument
199 def setup_predict_data_feeder(x, batch_size=None): argument
300 batch_size=None, argument
422 def batch_size(self): member in DataFeeder
591 def __init__(self, x, y, n_classes, batch_size): argument
808 batch_size, argument
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_decoder.h | 43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder()
60 int batch_size() { return batch_size_; } in batch_size() function
74 CTCGreedyDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCGreedyDecoder()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_decoder.h | 43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder()
60 int batch_size() { return batch_size_; } in batch_size() function
74 CTCGreedyDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCGreedyDecoder()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_test.py | 68 batch_size, argument
347 batch_size, argument
387 def test_training(self, num_units, input_size, batch_size, time, num_layers, argument
411 def test_training_fp16(self, num_units, input_size, batch_size, time, argument
438 def test_inference(self, num_units, input_size, batch_size, time, num_layers, argument
470 def test_inference_fp16(self, num_units, input_size, batch_size, time, argument
507 def test_inference_with_dropout(self, num_units, input_size, batch_size, time, argument
555 batch_size, argument
729 batch_size, argument
769 def test_training(self, num_units, input_size, batch_size, time, num_layers, argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | batch_features.h | 33 explicit BatchFeatures(int64 batch_size) : batch_size_(batch_size) {} in BatchFeatures()
75 int64 batch_size() const { return batch_size_; } in batch_size() function
|
| /external/tensorflow/tensorflow/contrib/rnn/ops/ |
| D | lstm_ops.cc | 50 DimensionHandle batch_size = c->Dim(x, 0); in __anon4050c6010102() local
137 DimensionHandle batch_size = c->Dim(x, 0); in __anon4050c6010202() local
206 DimensionHandle batch_size = c->Dim(x, 1); in __anon4050c6010302() local
|
| D | gru_ops.cc | 40 DimensionHandle batch_size = c->Dim(x, 0); in __anon07a5ee8f0102() local
119 DimensionHandle batch_size = c->Dim(x, 0); in __anon07a5ee8f0202() local
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | debug.py | 209 def predict_classes(self, input_fn=None, batch_size=None): argument
227 batch_size=None): argument
324 def predict_scores(self, input_fn=None, batch_size=None): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 85 def zero_state(self, batch_size, dtype): argument
103 def zero_state(self, batch_size, dtype): argument
122 def zero_state(self, batch_size, dtype): argument
340 def _assert_cell_builds(self, cell_class, dtype, batch_size, in_size, argument
811 def static_vs_dynamic_rnn_benchmark(batch_size, max_time, num_units, use_gpu): argument
873 def half_seq_len_vs_unroll_half_rnn_benchmark(batch_size, max_time, num_units, argument
941 def concat_state_vs_tuple_state_rnn_benchmark(batch_size, max_time, num_units, argument
1007 def dynamic_rnn_swap_memory_benchmark(batch_size, max_time, num_units): argument
1041 def rnn_long_sequence_benchmark(batch_size, seqlen, num_units, dynamic, argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_layer.py | 54 batch_size=None, argument
148 batch_size=None, argument
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
| D | mnist_graph_test.py | 41 def _create_graph(self, batch_size): argument
78 def _report(self, test_name, start, num_iters, batch_size): argument
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | gru_ops_test.py | 335 def training_gru_block_vs_gru_cell(batch_size, argument
410 def inference_gru_block_vs_gru_cell(batch_size, argument
468 def single_bprop_step_gru_block_vs_gru_cell(batch_size, argument
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_and_batch_benchmark.py | 116 def name(method, label, num_calls, inter_op, element_size, batch_size): argument
129 def make_dataset(element_size, num_calls, batch_size): # pylint: disable=missing-docstring argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | make_csv_dataset_test.py | 40 def _make_csv_dataset(self, filenames, batch_size, num_epochs=1, **kwargs): argument
64 def _next_expected_batch(self, expected_output, expected_keys, batch_size, argument
80 batch_size, argument
112 batch_size=1, argument
|
| D | make_tf_record_dataset_test.py | 34 def _read_test(self, batch_size, num_epochs, file_index=None, argument
103 def _shuffle_test(self, batch_size, num_epochs, num_parallel_reads=1, argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | batch_dataset_serialization_test.py | 33 def build_dataset(self, multiplier=15.0, tensor_slice_len=2, batch_size=2): argument
68 def _build_dataset_sparse(self, batch_size=5): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_split_op_test.cc | 79 void BM_StringSplit(int iters, int batch_size) { in BM_StringSplit()
110 void BM_StringSplitV2(int iters, int batch_size) { in BM_StringSplitV2()
|
| D | matrix_band_part_op_gpu.cu.cc | 33 const int batch_size, const int m, in MatrixBandPartKernel()
57 const int batch_size = input.dimension(0); in operator ()() local
|