| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 168 std::unique_ptr<OpKernel> op(CreateOpKernel( in ExpectSuccess()
185 CreateOpKernel(std::move(device_type), &device_, cpu_allocator(), in ExpectFailure()
364 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F()
448 std::unique_ptr<OpKernel> op(CreateOpKernel( in TEST_F()
512 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
546 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device, in ExpectFailure()
736 std::unique_ptr<OpKernel> op(CreateOpKernel( in TEST_F()
1011 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_InputRangeHelper()
1086 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_TraceString()
|
| D | op_kernel.cc | 1576 std::unique_ptr<OpKernel> CreateOpKernel( in CreateOpKernel() function
1588 return CreateOpKernel(device_type, device, allocator, props, in CreateOpKernel()
1592 std::unique_ptr<OpKernel> CreateOpKernel( in CreateOpKernel() function
1597 *status = CreateOpKernel(std::move(device_type), device, allocator, in CreateOpKernel()
1602 Status CreateOpKernel(DeviceType device_type, DeviceBase* device, in CreateOpKernel() function
1606 return CreateOpKernel(std::move(device_type), device, allocator, flib, in CreateOpKernel()
1611 Status CreateOpKernel(DeviceType device_type, DeviceBase* device, in CreateOpKernel() function
|
| D | op_segment_test.cc | 67 auto created = CreateOpKernel(DEVICE_CPU, &device_, cpu_allocator(), in GetFn()
|
| D | resource_op_kernel_test.cc | 100 std::unique_ptr<OpKernel> op(CreateOpKernel( in CreateOp()
|
| D | op_kernel.h | 1375 std::unique_ptr<OpKernel> CreateOpKernel(DeviceType device_type,
1381 std::unique_ptr<OpKernel> CreateOpKernel(
1386 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
1391 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 72 CreateOpKernel("CPU", cpu_device, cpu_device->GetAllocator({}), node, in EvaluateNode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op_test.cc | 64 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, device.get(), in PersistentMemoryTrackingTest()
|
| D | restore_op_test.cc | 91 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()
|
| D | segment_reduction_ops_test.cc | 68 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), in BM_SegmentReduction()
|
| D | parse_tensor_test.cc | 53 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in ParseSerializedWithNodeDef()
|
| D | nn_ops_test.cc | 786 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_LRNFloat()
866 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_AvgPool()
970 CreateOpKernel(DEVICE_CPU, nullptr, cpu_allocator(), in BM_AvgPoolBk()
1060 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_MaxPool()
1235 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_ReluFloat()
1307 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), in BM_SoftplusFloat()
|
| D | sparse_to_dense_op_test.cc | 244 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_SparseToDense()
|
| D | ops_testutil.cc | 146 TF_RETURN_IF_ERROR(CreateOpKernel( in InitOpWithGraphVersion()
|
| D | restore_v2_op_test.cc | 103 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), save, in RunTest()
|
| D | collective_nccl_test.cc | 54 std::unique_ptr<OpKernel> k = CreateOpKernel( in GetKernel()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 56 CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status); in TestBitcastOp()
|
| D | summary_op_test.cc | 71 CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status); in TestScalarSummaryOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 144 status.Update(CreateOpKernel(device_type_, device_, allocator(), flr, in InitOpWithFunctionLibrary()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op_test.cc | 102 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, map_defun_kernel)); in CreateMapDefunOpKernel()
|
| D | dataset_test_base.cc | 303 Status DatasetOpsTestBase::CreateOpKernel( in CreateOpKernel() function in tensorflow::data::DatasetOpsTestBase
311 TF_RETURN_IF_ERROR(tensorflow::CreateOpKernel( in CreateOpKernel()
906 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, dataset_kernel)); in MakeDatasetOpKernel()
|
| D | dataset_test_base.h | 656 Status CreateOpKernel(const NodeDef& node_def,
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 118 return CreateOpKernel(DeviceType(device_name), nullptr, nullptr, def, 1, in GetFakeKernel()
206 return CreateOpKernel(DeviceType("FakeDevice"), nullptr, nullptr, def, 1, in GetFakeKernelWithAttr()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 244 return CreateOpKernel("GPU", gpu_helper_->gpu(), in GetOpKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer_test.cc | 105 std::unique_ptr<OpKernel> k = CreateOpKernel( in GetKernel()
|
| D | ring_reducer_test.cc | 106 std::unique_ptr<OpKernel> k = CreateOpKernel( in GetKernel()
|