Home
last modified time | relevance | path

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

/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()
Deigen_spatial_convolutions_test.cc894 const int batches = 2; in TEST() local
960 const int batches = 2; in TEST() local
/external/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_batch_cache.h50 struct fd_batch *batches[32]; member
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
Dreference_ops.h167 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Conv() local
263 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Conv() local
457 const int batches = ArraySize(output_dims, 1) * ArraySize(output_dims, 2) * in FullyConnected() local
509 const int batches = ArraySize(output_dims, 1) * ArraySize(output_dims, 2) * in FullyConnected() local
569 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in NonGlobalBatchNormalization() local
604 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in GlobalBatchNormalization() local
628 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Relu() local
648 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Relu1() local
669 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Relu6() local
692 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in L2Normalization() local
[all …]
Ddepthwiseconv_float.h33 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in DepthwiseConv() local
Ddepthwiseconv_uint8.h39 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in DepthwiseConv() local
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dsoftmax_test.cc34 SoftmaxOpModel(int batches, int size, float beta) in SoftmaxOpModel()
Dbasic_rnn_test.cc127 RNNOpModel(int batches, int units, int size) in RNNOpModel()
Dunidirectional_sequence_rnn_test.cc127 UnidirectionalRNNOpModel(int batches, int sequence_len, int units, int size, in UnidirectionalRNNOpModel()
Dsvdf_test.cc131 SVDFOpModel(int batches, int units, int input_size, int memory_size, int rank) in SVDFOpModel()
Dfully_connected_test.cc122 BaseFullyConnectedOpModel(int units, int batches, const TensorData& input, in BaseFullyConnectedOpModel()
Ddepthwise_conv.cc126 int batches = SizeOfDimension(input, 0); in Prepare() local
Dbidirectional_sequence_rnn_test.cc656 BidirectionalRNNOpModel(int batches, int sequence_len, int fw_units, in BidirectionalRNNOpModel()
Dpooling.cc76 int batches = input->dims->data[0]; in GenericPrepare() local
Dconv.cc157 int batches = input->dims->data[0]; in Prepare() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Dmultithreaded_conv.h172 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Conv() local
Doptimized_ops.h575 const int batches = ArraySize(output_dims, 1) * ArraySize(output_dims, 2) * in FullyConnected() local
729 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Im2col() local
1109 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in NonGlobalBatchNormalization() local
1145 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in GlobalBatchNormalization() local
1179 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Relu1() local
1201 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Relu6() local
1225 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in L2Normalization() local
1296 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in L2Normalization() local
1908 const int batches = in Div() local
1981 const int batches = in Sub() 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/python/ops/
Ddata_flow_ops.py2316 batches=None, argument