| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | variant_ops_util.cc | 38 std::function<Status(OpKernelContext*, const Variant&, const Variant&, in AddVariantTo()
56 std::function<Status(OpKernelContext*, const Variant&, in AddNVariant()
|
| H A D | meta_support.h | 29 class OpKernelContext; variable
|
| H A D | save_restore_tensor.h | 24 class OpKernelContext; variable
|
| H A D | conv_ops.h | 32 class OpKernelContext; variable
|
| H A D | random_binomial_op.h | 24 class OpKernelContext; variable
|
| H A D | random_op.h | 24 class OpKernelContext; variable
|
| H A D | variant_ops_util.h | 25 class OpKernelContext; variable
|
| H A D | reshape_util.h | 25 class OpKernelContext; variable
|
| H A D | parameterized_truncated_normal_op.h | 25 class OpKernelContext; variable
|
| H A D | tensor_list_util.h | 24 class OpKernelContext; variable
|
| H A D | conv_grad_ops.h | 170 class OpKernelContext; variable
|
| H A D | xsmm_conv2d.h | 27 class OpKernelContext; variable
|
| H A D | scatter_nd_op.h | 35 class OpKernelContext; variable
|
| H A D | scatter_functor.h | 34 class OpKernelContext; variable
|
| H A D | segment_reduction_ops.h | 26 class OpKernelContext; variable
|
| H A D | gather_nd_op.h | 30 class OpKernelContext; variable
|
| H A D | scatter_nd_op_cpu_impl.h | 42 class OpKernelContext; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.h | 48 class OpKernelContext; variable
223 virtual Status ReinitializeGpuDevice(OpKernelContext* /*context*/, in ReinitializeGpuDevice() argument
|
| H A D | ref_var.h | 22 class OpKernelContext; variable
|
| H A D | lookup_interface.h | 25 class OpKernelContext; variable
|
| H A D | ref_var.cc | 25 std::function<void(OpKernelContext*, Tensor*, const Tensor&)> copy) { in AssignRefVariable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | blas_gemm.h | 29 class OpKernelContext; variable
|
| H A D | gru_ops.h | 26 class OpKernelContext; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_tpu_device.cc | 51 OpKernelContext* /*unused ctx*/, int device_ordinal, in InitScopedRef() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | step_stats_collector.h | 36 class OpKernelContext; variable
|