Home
last modified time | relevance | path

Searched defs:batches (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_generator_test.py339 def _make_dataset(self, inputs, batches): argument
342 def _make_iterator(self, inputs, batches): argument
346 def _make_generator(self, inputs, batches): argument
/external/tensorflow/tensorflow/core/kernels/
Dadjust_contrast_op_benchmark_test.cc24 static Graph* BM_AdjustContrast(int batches, int width, int height) { in BM_AdjustContrast()
Dresize_op_benchmark_test.cc24 static Graph* BM_Resize(const char* algorithm, int batches, int width, in BM_Resize()
Dcrop_and_resize_op_benchmark_test.cc24 static Graph* BM_CropAndResize(int batches, int width, int height, int depth, in BM_CropAndResize()
/external/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_batch_cache.h50 struct fd_batch *batches[32]; member
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h44 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnected() local
90 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
138 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
192 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected() local
Ddepthwiseconv_float.h44 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConv() local
Ddepthwiseconv_uint8.h118 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Run() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dpooling.h31 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
92 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
Dfully_connected.h42 const int batches = output_shape.Dims(0); in FullyConnected() local
Dconv.h50 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in ConvPerChannel() local
Ddepthwise_conv.h50 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvPerChannel() local
/external/mesa3d/src/mesa/main/
Dglthread.h79 struct glthread_batch batches[MARSHAL_MAX_BATCHES]; member
/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn_test.cc176 RNNOpModel(int batches, int units, int size, in RNNOpModel()
236 HybridRNNOpModel(int batches, int units, int size, TensorType tensor_type) in HybridRNNOpModel()
Dlog_softmax_test.cc34 LogSoftmaxOpModel(int batches, int size) in LogSoftmaxOpModel()
Dsvdf_test.cc131 BaseSVDFOpModel(int batches, int units, int input_size, int memory_size, in BaseSVDFOpModel()
205 HybridSVDFOpModel(int batches, int units, int input_size, int memory_size, in HybridSVDFOpModel()
Dunidirectional_sequence_rnn_test.cc175 int batches, int sequence_len, int units, int size, bool time_major, in UnidirectionalRNNOpModel()
250 HybridUnidirectionalRNNOpModel(int batches, int sequence_len, int units, in HybridUnidirectionalRNNOpModel()
Dsoftmax_test.cc34 SoftmaxOpModel(int batches, int size, float beta) in SoftmaxOpModel()
Dfully_connected_test.cc137 TfLiteRegistration* registration, int units, int batches, in BaseFullyConnectedOpModel()
276 HybridFullyConnectedOpModel(int units, int batches, const TensorData& input, in HybridFullyConnectedOpModel()
550 int batches, FullyConnectedOptionsWeightsFormat weights_format) { in SimpleTestQuantizedInt16OutputCase()
Ddepthwise_conv.cc131 int batches = SizeOfDimension(input, 0); in Prepare() local
Dpooling.cc76 int batches = input->dims->data[0]; in GenericPrepare() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h153 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv() local
Doptimized_ops.h1101 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnectedAsGEMV() local
1189 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
1254 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
1319 const int8* shuffled_weights_data, int batches, int output_depth, in ShuffledFullyConnectedWorkerImpl()
1656 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected() local
2083 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DilatedIm2col() local
2157 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col() local
3971 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
4043 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
4163 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
[all …]
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/
Dgraph_functions_wrapper.c213 const uint32_t nn_id, const uint32_t batches, const uint32_t height, in hexagon_controller_ExecuteGraph()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1172 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
1262 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected() local
1375 const int8* shuffled_weights_data, int batches, int output_depth, in ShuffledFullyConnectedWorkerImpl()
1712 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected() local
1960 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DilatedIm2col() local
2058 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col() local
4118 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
4182 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool() local
4294 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
4350 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool() local
[all …]

12