| /external/tensorflow/tensorflow/core/kernels/data/ |
| dataset_test_base.cc | 52 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, tensor_slice_dataset_kernel));
89 Status DatasetOpsTestBase::CreateOpKernel(
92 TF_RETURN_IF_ERROR(tensorflow::CreateOpKernel(device_type_, device_.get(),
|
| dataset_test_base.h | 64 Status CreateOpKernel(const NodeDef& node_def,
82 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, range_op_kernel));
|
| concatenate_dataset_op_test.cc | 54 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
|
| zip_dataset_op_test.cc | 64 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
|
| map_dataset_op_test.cc | 61 TF_CHECK_OK(CreateOpKernel(map_node_def_, map_kernel));
|
| repeat_dataset_op_test.cc | 47 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
|
| sparse_tensor_slice_dataset_op_test.cc | 45 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
|
| tensor_dataset_op_test.cc | 51 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, tensor_dataset_kernel));
|
| /external/tensorflow/tensorflow/c/kernels/ |
| bitcast_op_test.cc | 57 CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| evaluation_utils.cc | 93 CreateOpKernel("CPU", cpu_device, cpu_device->GetAllocator({}), node,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| constant_op_test.cc | 60 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, device.get(),
|
| ops_testutil.h | 104 kernel_ = CreateOpKernel(device_type_, device_.get(), allocator(),
|
| parse_tensor_test.cc | 50 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
|
| segment_reduction_ops_test.cc | 69 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(),
|
| sparse_to_dense_op_test.cc | 246 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
|
| restore_op_test.cc | 88 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
392 std::unique_ptr<OpKernel> op(CreateOpKernel(DEVICE_CPU, device.get(),
|
| collective_nccl_reducer_test.cc | 45 std::unique_ptr<OpKernel> k = CreateOpKernel(
|
| collective_ops.cc | 174 k = CreateOpKernel(c->device_type(), c->device(),
|
| restore_v2_op_test.cc | 100 CreateOpKernel(DEVICE_CPU, device.get(), cpu_allocator(), save,
|
| nn_ops_test.cc | [all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel_test.cc | 160 std::unique_ptr<OpKernel> op(CreateOpKernel(
177 CreateOpKernel(std::move(device_type), &device_, cpu_allocator(),
328 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(),
353 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(),
381 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(),
436 // Test CreateOpKernel()
437 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
470 // Test CreateOpKernel().
471 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
663 std::unique_ptr<OpKernel> op(CreateOpKernel(
[all...] |
| op_segment_test.cc | 67 auto created = CreateOpKernel(DEVICE_CPU, &device_, cpu_allocator(),
|
| resource_op_kernel_test.cc | 99 std::unique_ptr<OpKernel> op(CreateOpKernel(
|
| op_kernel.cc | [all...] |
| /external/tensorflow/tensorflow/c/ |
| kernels_test.cc | 82 return CreateOpKernel(DeviceType(device_name), nullptr, nullptr, def, 1,
|
