Searched refs:VectorBatchVectorCwiseProduct (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 208 tensor_utils::VectorBatchVectorCwiseProduct( in LstmStepWithAuxInput()
229 tensor_utils::VectorBatchVectorCwiseProduct( in LstmStepWithAuxInput()
246 tensor_utils::VectorBatchVectorCwiseProduct( in LstmStepWithAuxInput()
279 tensor_utils::VectorBatchVectorCwiseProduct( in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_common.h | 368 inline void VectorBatchVectorCwiseProduct(const T* vector, int v_size, in VectorBatchVectorCwiseProduct() function
|
| D | tensor_utils_test.cc | 1884 VectorBatchVectorCwiseProduct(input, kVectorSize, output.data(), kBatchSize, in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 89 tensor_utils::VectorBatchVectorCwiseProduct(layer_norm_coefficients, n_cell, in CalculateLstmGateFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 194 tensor_utils::VectorBatchVectorCwiseProduct(layer_norm_coefficients, n_cell, in CalculateLstmGateFloat()
405 tensor_utils::VectorBatchVectorCwiseProduct(layer_norm_coefficients, n_cell, in CalculateLstmGateHybrid()
|