Searched refs:BatchFunctionKernel (Results 1 – 5 of 5) sorted by relevance
237 BatchFunctionKernel::BatchFunctionKernel(OpKernelConstruction* c) in BatchFunctionKernel() function in tensorflow::BatchFunctionKernel286 bool BatchFunctionKernel::IsExpensive() { return false; } in IsExpensive()288 void BatchFunctionKernel::ComputeAsync(OpKernelContext* c, DoneCallback done) { in ComputeAsync()363 Status BatchFunctionKernel::InstantiateFunction( in InstantiateFunction()423 Status BatchFunctionKernel::GetOrCreateFunctionHandle( in GetOrCreateFunctionHandle()439 Status BatchFunctionKernel::ValidateAllowedBatchSizes() const { in ValidateAllowedBatchSizes()470 void BatchFunctionKernel::SetAdaptiveBatchSchedulerOptions( in SetAdaptiveBatchSchedulerOptions()523 BatchFunctionKernel);531 BatchFunctionKernel);537 BatchFunctionKernel);
27 BatchFunctionKernel* batch_kernel = in TEST_P()28 dynamic_cast<BatchFunctionKernel*>(op_kernel()); in TEST_P()
31 explicit BatchFunctionKernelTestAccess(BatchFunctionKernel* kernel);36 BatchFunctionKernel* const kernel_;
49 class BatchFunctionKernel : public AsyncOpKernel {51 explicit BatchFunctionKernel(OpKernelConstruction* c);
22 BatchFunctionKernel* kernel) in BatchFunctionKernelTestAccess()