| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 160 std::unique_ptr<OpKernel> op(CreateOpKernel( in ExpectSuccess()
177 CreateOpKernel(std::move(device_type), &device_, cpu_allocator(), in ExpectFailure()
328 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F()
353 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F()
381 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F()
437 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
471 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device, in ExpectFailure()
663 std::unique_ptr<OpKernel> op(CreateOpKernel( in TEST_F()
938 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_InputRangeHelper()
|
| D | op_segment_test.cc | 67 auto created = CreateOpKernel(DEVICE_CPU, &device_, cpu_allocator(), in GetFn()
|
| D | resource_op_kernel_test.cc | 99 std::unique_ptr<OpKernel> op(CreateOpKernel( in CreateOp()
|
| D | op_kernel.cc | 1256 std::unique_ptr<OpKernel> CreateOpKernel( in CreateOpKernel() function
1260 *status = CreateOpKernel(std::move(device_type), device, allocator, nullptr, in CreateOpKernel()
1265 Status CreateOpKernel(DeviceType device_type, DeviceBase* device, in CreateOpKernel() function
|
| D | op_kernel.h | 1314 std::unique_ptr<OpKernel> CreateOpKernel(DeviceType device_type,
1319 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 52 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, tensor_slice_dataset_kernel)); in CreateTensorSliceDatasetKernel()
89 Status DatasetOpsTestBase::CreateOpKernel( in CreateOpKernel() function in tensorflow::data::DatasetOpsTestBase
92 TF_RETURN_IF_ERROR(tensorflow::CreateOpKernel(device_type_, device_.get(), in CreateOpKernel()
|
| D | dataset_test_base.h | 64 Status CreateOpKernel(const NodeDef& node_def,
82 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, range_op_kernel)); in CreateRangeDatasetOpKernel()
|
| D | concatenate_dataset_op_test.cc | 54 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateConcatenateDatasetKernel()
|
| D | zip_dataset_op_test.cc | 64 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateZipDatasetKernel()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 45 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateSparseTensorSliceDatasetKernel()
|
| D | tensor_dataset_op_test.cc | 51 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, tensor_dataset_kernel)); in CreateTensorDatasetKernel()
|
| D | tensor_slice_dataset_op_test.cc | 52 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, tensor_dataset_kernel)); in CreateTensorSliceDatasetKernel()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 57 CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status); in TestBitcastOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op_test.cc | 60 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, device.get(), in PersistentMemoryTrackingTest()
|
| D | restore_op_test.cc | 88 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in TEST_F()
392 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in TEST_F()
|
| D | segment_reduction_ops_test.cc | 69 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), in BM_SegmentReduction()
|
| D | parse_tensor_test.cc | 50 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in ParseSerializedWithNodeDef()
|
| D | ops_testutil.h | 104 kernel_ = CreateOpKernel(device_type_, device_.get(), allocator(), in InitOpWithGraphVersion()
|
| D | sparse_to_dense_op_test.cc | 246 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_SparseToDense()
|
| D | nn_ops_test.cc | 760 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_LRNFloat()
841 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_AvgPool()
947 CreateOpKernel(DEVICE_CPU, nullptr, cpu_allocator(), in BM_AvgPoolBk()
1036 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_MaxPool()
1212 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(), in BM_ReluFloat()
|
| D | restore_v2_op_test.cc | 100 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), save, in RunTest()
|
| D | collective_nccl_reducer_test.cc | 45 std::unique_ptr<OpKernel> k = CreateOpKernel( in GetKernel()
|
| D | collective_ops.cc | 174 k = CreateOpKernel(c->device_type(), c->device(), in BuildOpKernel()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 93 CreateOpKernel("CPU", cpu_device, cpu_device->GetAllocator({}), node, in EvaluateNode()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 82 return CreateOpKernel(DeviceType(device_name), nullptr, nullptr, def, 1, in GetFakeKernel()
|