Searched refs:num_batch (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 138 int num_batch, OpKernelContext* ctx, in SetTrtEngineInputs() argument
153 if (num_batch != input_shape.dim_size(0)) { in SetTrtEngineInputs()
155 StrCat("Input data has inconsistent batch size: ", num_batch, in SetTrtEngineInputs()
|
| D | trt_engine_utils.h | 138 int num_batch, OpKernelContext* ctx = nullptr,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 781 const int num_batch = in ExecuteTrtEngine() local
786 use_implicit_batch_, num_batch, ctx)); in ExecuteTrtEngine()
790 use_implicit_batch_, num_batch, ctx)); in ExecuteTrtEngine()
800 use_implicit_batch_, num_batch)); in ExecuteTrtEngine()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | lstm_v2_test.py | 1095 num_batch = 10
1107 train_samples=(batch * num_batch),
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1473 int64_t num_batch = input_type.getNumElements() / num_elems; in matchAndRewrite() local
1474 SmallVector<int64_t, 2> shape_vals({num_batch, num_elems}); in matchAndRewrite()
|