| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_test.py | 67 input_size, argument
346 input_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
554 input_size, argument
728 input_size, argument
769 def test_training(self, num_units, input_size, batch_size, time, num_layers, argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | gru_ops_test.py | 337 input_size, argument
412 input_size, argument
470 input_size, argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | conv_grad_size_util.cc | 25 int64 input_size, int64 filter_size, int64 dilation_rate, int64 stride, in GetWindowedOutputSize()
67 ConvGradExtractAndVerifyDimension(int64 input_size, int64 filter_size, in ConvGradExtractAndVerifyDimension()
|
| D | quantize.h | 52 const int64 input_size = input.size(); in PackToUint32() local
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax.cc | 73 const int input_size = input->dims->data[0]; in Softmax1DFloat() local
82 const int input_size = input->dims->data[1]; in Softmax2DFloat() local
93 const int input_size = input->dims->data[0]; in Softmax1DQuantized() local
112 const int input_size = input->dims->data[1]; in Softmax2DQuantized() local
|
| /external/eigen/bench/tensors/ |
| D | tensor_benchmarks.h | 114 Eigen::array<TensorIndex, 2> input_size; in rowChip() local
131 Eigen::array<TensorIndex, 2> input_size; in colChip() local
312 Eigen::array<TensorIndex, 2> input_size; in rowReduction() local
340 Eigen::array<TensorIndex, 2> input_size; in colReduction() local
370 Eigen::array<TensorIndex, 2> input_size; in fullReduction() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 24 int input_size, int num_units, int batch_size, in RnnBatchStep()
40 int input_size, int aux_input_size, int num_units, in RnnBatchStep()
123 float recurrent_weights_scale, const float* bias_ptr, int input_size, in RnnBatchStep()
145 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/micro_features/ |
| D | fft_util.cc | 24 struct FftState* state, size_t input_size) { in FftPopulateState()
|
| D | fft.cc | 28 const size_t input_size = state->input_size; in FftCompute() local
|
| D | fft.h | 32 size_t input_size; member
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 533 const int input_size = in SigmoidEval() local
549 void Softmax(const float* in, const int input_size, const int batch_size, in Softmax()
583 const int input_size = input->dims->data[0]; in Softmax1DFloat() local
591 const int input_size = input->dims->data[1]; in Softmax2DFloat() local
600 const int input_size = input->dims->data[2]; in Softmax3DFloat() local
616 const int input_size = input->dims->data[0]; in Softmax1DQuantizedUint8() local
633 const int input_size = input->dims->data[1]; in Softmax2DQuantizedUint8() local
649 const int input_size = input->dims->data[2]; in Softmax3DQuantizedUint8() local
686 const int input_size = input->dims->data[0]; in Softmax1DQuantizedInt8() local
700 const int input_size = input->dims->data[1]; in Softmax2DQuantizedInt8() local
[all …]
|
| D | softmax_test.cc | 81 const int input_size = 5; in TEST() local
110 const int input_size = 5; in TEST() local
|
| D | svdf_test.cc | 131 BaseSVDFOpModel(int batches, int units, int input_size, int memory_size, in BaseSVDFOpModel()
179 int input_size() { return input_size_; } in input_size() function in tflite::__anond1a0a9e40111::BaseSVDFOpModel
205 HybridSVDFOpModel(int batches, int units, int input_size, int memory_size, in HybridSVDFOpModel()
|
| /external/pdfium/testing/ |
| D | test_support.h | 38 uint32_t input_size; member
45 uint32_t input_size; member
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/lib/ |
| D | fft_util.c | 23 int FftPopulateState(struct FftState* state, size_t input_size) { in FftPopulateState()
|
| D | fft.c | 28 const size_t input_size = state->input_size; in FftCompute() local
|
| D | fft.h | 34 size_t input_size; member
|
| /external/libchrome/base/ |
| D | base64.cc | 29 size_t input_size = input.size(); in Base64Decode() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | string_ops.cc | 310 DimensionHandle input_size = c->NumElements(c->input(0)); in __anonb4ab91930902() local
332 DimensionHandle input_size = c->NumElements(c->input(0)); in __anonb4ab91930a02() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops_kernel_argmax_float_1d.cc | 29 int64 input_size = *static_cast<int64*>(data[1]); in argmax_float_1d_xla_impl() local
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | graph_placer_test.py | 38 input_size=256, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | tanh.h | 26 int32_t input_size, const int8_t* input_data, in Tanh()
|
| D | logistic.h | 26 int32_t input_size, const int8_t* input_data, in Logistic()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_pack.cc | 42 int input_size = RequiredBufferSizeForShape(input_array.shape()); in Pack() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_slice_op.cc | 35 const Tensor& input_size = context->input(4); in Compute() local
|