Home
last modified time | relevance | path

Searched defs:OpKernelContext (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dreshape_util.h24 class OpKernelContext; variable
Dsave_restore_tensor.h24 class OpKernelContext; variable
Dmeta_support.h29 class OpKernelContext; variable
Dparameterized_truncated_normal_op.h24 class OpKernelContext; variable
Dconv_ops.h32 class OpKernelContext; variable
Drandom_op.h24 class OpKernelContext; variable
Drandom_binomial_op.h24 class OpKernelContext; variable
Dconv_grad_ops.h170 class OpKernelContext; variable
Dxsmm_conv2d.h27 class OpKernelContext; variable
Dscatter_nd_op.h36 class OpKernelContext; variable
Dscatter_functor.h33 class OpKernelContext; variable
Dscatter_nd_op_cpu_impl.h45 class OpKernelContext; variable
Dgather_nd_op.h30 class OpKernelContext; variable
Dsegment_reduction_ops.h33 class OpKernelContext; variable
Ddepthwise_conv_op.h60 class OpKernelContext; variable
Ddeep_conv2d.h23 class OpKernelContext; variable
Dinplace_ops.cc151 void Compute(OpKernelContext*) override {} in Compute() argument
/external/tensorflow/tensorflow/core/framework/
Dlookup_interface.h25 class OpKernelContext; variable
Ddevice_base.h50 class OpKernelContext; variable
229 virtual Status ReinitializeGpuDevice(OpKernelContext* /*context*/, in ReinitializeGpuDevice() argument
Dvariant_op_registry.h36 class OpKernelContext; variable
/external/tensorflow/tensorflow/core/kernels/rnn/
Dblas_gemm.h29 class OpKernelContext; variable
Dgru_ops.h26 class OpKernelContext; variable
Dlstm_ops.h26 class OpKernelContext; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h37 class OpKernelContext; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.h34 class OpKernelContext; variable

12