Searched refs:num_batch (Results 1 – 5 of 5) sorted by relevance
| /external/ComputeLibrary/tests/validation/reference/ |
| D | QLSTMLayerNormalization.cpp | 57 const uint32_t num_batch = src.shape()[1]; in qlstm_layer_normalization() local
60 for(uint32_t batch_idx = 0; batch_idx < num_batch; ++batch_idx) in qlstm_layer_normalization()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 119 int num_batch, in SetTrtEngineInputs() argument
152 if (num_batch != input_shape.dim_size(0)) { in SetTrtEngineInputs()
154 StrCat("Input data has inconsistent batch size: ", num_batch, in SetTrtEngineInputs()
|
| D | trt_engine_utils.h | 47 int num_batch,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 983 const int num_batch = in ExecuteTrtEngine() local
988 use_implicit_batch_, num_batch, profiles, ctx)); in ExecuteTrtEngine()
992 use_implicit_batch_, num_batch, ctx)); in ExecuteTrtEngine()
1014 num_batch); in ExecuteTrtEngine()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1501 int64_t num_batch = input_type.getNumElements() / num_elems; in matchAndRewrite() local
1502 SmallVector<int64_t, 2> shape_vals({num_batch, num_elems}); in matchAndRewrite()
|