Home
last modified time | relevance | path

Searched refs:CreateOpKernel (Results 1 – 25 of 33) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel_test.cc171 std::unique_ptr<OpKernel> op(CreateOpKernel( in ExpectSuccess()
188 CreateOpKernel(std::move(device_type), &device_, cpu_allocator(), in ExpectFailure()
367 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F()
451 std::unique_ptr<OpKernel> op(CreateOpKernel( in TEST_F()
622 std::unique_ptr<OpKernel> op(CreateOpKernel( in TEST_F()
908 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_InputRangeHelper()
983 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_TraceString()
H A Dop_kernel_test_base.h86 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
120 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device, in ExpectFailure()
H A Dop_kernel.cc1644 std::unique_ptr<OpKernel> CreateOpKernel( in CreateOpKernel() function
1656 return CreateOpKernel(device_type, device, allocator, props, in CreateOpKernel()
1660 std::unique_ptr<OpKernel> CreateOpKernel( in CreateOpKernel() function
1665 *status = CreateOpKernel(std::move(device_type), device, allocator, in CreateOpKernel()
1670 Status CreateOpKernel(DeviceType device_type, DeviceBase* device, in CreateOpKernel() function
1674 return CreateOpKernel(std::move(device_type), device, allocator, flib, in CreateOpKernel()
1679 Status CreateOpKernel(DeviceType device_type, DeviceBase* device, in CreateOpKernel() function
H A Dop_kernel.h1344 std::unique_ptr<OpKernel> CreateOpKernel(DeviceType device_type,
1350 std::unique_ptr<OpKernel> CreateOpKernel(
1355 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
1360 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
H A Dop_segment_test.cc67 auto created = CreateOpKernel(DEVICE_CPU, &device_, cpu_allocator(), in GetFn()
H A Dresource_op_kernel_test.cc100 std::unique_ptr<OpKernel> op(CreateOpKernel( in CreateOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dop_kernel_runner.cc67 tensorflow::Status CreateOpKernel( in CreateOpKernel() function
127 TF_RETURN_IF_ERROR(CreateOpKernel(function_library_runtime, in Create()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dop_kernel_runner.cc69 tensorflow::Status CreateOpKernel( in CreateOpKernel() function
124 TF_RETURN_IF_ERROR(CreateOpKernel(function_library_runtime, in Create()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_test.cc72 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), in BM_UnsortedSegmentReduction()
136 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), in BM_SegmentReduction()
H A Dnn_ops_test.cc801 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_LRNFloat()
881 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_AvgPool()
985 CreateOpKernel(DEVICE_CPU, nullptr, cpu_allocator(), in BM_AvgPoolBk()
1075 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_MaxPool()
1257 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_ReluFloat()
1329 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), in BM_SoftplusFloat()
H A Drestore_op_test.cc91 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in TEST_F()
395 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in TEST_F()
H A Dconstant_op_test.cc64 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, device.get(), in PersistentMemoryTrackingTest()
H A Dparse_tensor_test.cc53 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in ParseSerializedWithNodeDef()
H A Dsparse_to_dense_op_test.cc244 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_SparseToDense()
H A Dops_testutil.cc146 TF_RETURN_IF_ERROR(CreateOpKernel( in InitOpWithGraphVersion()
H A Drestore_v2_op_test.cc103 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), save, in RunTest()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.cc298 Status DatasetOpsTestBase::CreateOpKernel( in CreateOpKernel() function in tensorflow::data::DatasetOpsTestBase
315 TF_RETURN_IF_ERROR(tensorflow::CreateOpKernel( in CreateOpKernel()
922 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, dataset_kernel)); in MakeDatasetOpKernel()
937 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, op_kernel)); in MakeGetOptionsOpKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Devaluation_utils.cc72 CreateOpKernel("CPU", cpu_device, cpu_device->GetAllocator({}), node, in EvaluateNode()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc114 std::unique_ptr<tensorflow::OpKernel> op_kernel = tensorflow::CreateOpKernel( in EvaluateOperation()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_input_task_test.cc118 batch_kernel_ = CreateOpKernel( in BatchInputTaskTest()
/aosp_15_r20/external/tensorflow/tensorflow/c/kernels/
H A Dbitcast_op_test.cc56 CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status); in TestBitcastOp()
H A Dsummary_op_test.cc71 CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status); in TestScalarSummaryOp()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels_test.cc120 return CreateOpKernel(DeviceType(device_name), nullptr, nullptr, def, 1, in GetFakeKernel()
146 return CreateOpKernel(DeviceType(device_name), nullptr, nullptr, def, 1, in GetFakeKernel2()
281 return CreateOpKernel(DeviceType("FakeDevice"), nullptr, nullptr, def, 1, in GetFakeKernelWithAttr()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmap_defun_op_test.cc102 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, map_defun_kernel)); in CreateMapDefunOpKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op_test.cc183 status.Update(CreateOpKernel(device_type_, device_, allocator(), flr, in InitOpWithFunctionLibrary()

12