Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dvariant_ops_util.cc38 std::function<Status(OpKernelContext*, const Variant&, const Variant&, in AddVariantTo()
56 std::function<Status(OpKernelContext*, const Variant&, in AddNVariant()
Dmeta_support.h29 class OpKernelContext; variable
Dsave_restore_tensor.h24 class OpKernelContext; variable
Drandom_op.h24 class OpKernelContext; variable
Drandom_binomial_op.h24 class OpKernelContext; variable
Dconv_ops.h32 class OpKernelContext; variable
Dtensor_list_util.h24 class OpKernelContext; variable
Dparameterized_truncated_normal_op.h25 class OpKernelContext; variable
Dvariant_ops_util.h25 class OpKernelContext; variable
Dreshape_util.h25 class OpKernelContext; variable
Dconv_grad_ops.h170 class OpKernelContext; variable
Dxsmm_conv2d.h27 class OpKernelContext; variable
Dscatter_nd_op.h35 class OpKernelContext; variable
Dscatter_functor.h34 class OpKernelContext; variable
Dsegment_reduction_ops.h26 class OpKernelContext; variable
Dgather_nd_op.h30 class OpKernelContext; variable
Dscatter_nd_op_cpu_impl.h42 class OpKernelContext; variable
/external/tensorflow/tensorflow/core/framework/
Dref_var.h22 class OpKernelContext; variable
Dlookup_interface.h25 class OpKernelContext; variable
Dref_var.cc25 std::function<void(OpKernelContext*, Tensor*, const Tensor&)> copy) { in AssignRefVariable()
Ddevice_base.h48 class OpKernelContext; variable
223 virtual Status ReinitializeGpuDevice(OpKernelContext* /*context*/, in ReinitializeGpuDevice() argument
/external/tensorflow/tensorflow/core/kernels/rnn/
Dblas_gemm.h29 class OpKernelContext; variable
Dgru_ops.h26 class OpKernelContext; variable
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_tpu_device.cc51 OpKernelContext* /*unused ctx*/, int device_ordinal, in InitScopedRef() argument
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h36 class OpKernelContext; variable

12