Searched refs:GetModelName (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 121 const string& GetModelName(OpKernelContext* ctx) { in GetModelName() function
178 RecordInputBatchSize(tensors[0].shape().dim_size(0), GetModelName(context), in RegisterInput()
181 batcher_queue_options_.batch_timeout_micros, GetModelName(context), in RegisterInput()
184 GetModelName(context), in RegisterInput()
187 batcher_queue_options_.max_enqueued_batches, GetModelName(context), in RegisterInput()
190 GetModelName(context), in RegisterInput()
331 RecordPaddingSize(padding_amount, GetModelName(context), padded_batch_size, in ConcatInputTensors()
333 RecordProcessedBatchSize(padded_batch_size, GetModelName(context), in ConcatInputTensors()
606 const string& model_name = GetModelName(last_task_context); in ProcessFuncBatch()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 73 const string& GetModelName(OpKernelContext* ctx) { in GetModelName() function
251 GetModelName(c)); in ComputeAsync()
253 RecordBatchParamNumBatchThreads(num_batch_threads_, GetModelName(c)); in ComputeAsync()
|