| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation_test.cc | 35 TF_ASSERT_OK(op->SetDeviceName("/device:DONTHAVE")); in TEST()
38 TF_ASSERT_OK(op->SetDeviceName("")); in TEST()
41 TF_ASSERT_OK(op->SetDeviceName("/job:localhost")); in TEST()
44 EXPECT_NE(Status::OK(), op->SetDeviceName("/not/a/valid/name")); in TEST()
|
| D | eager_operation.cc | 349 return SetDeviceName(device_name); in Reset()
449 Status EagerOperation::SetDeviceName(const char* c_name) { in SetDeviceName() function in tensorflow::EagerOperation
|
| D | mkl_eager_op_rewrite.cc | 162 return (*new_mkl_op)->SetDeviceName(device_name.c_str()); in SetupNewOp()
|
| D | eager_operation.h | 70 Status SetDeviceName(const char* name) override;
|
| /external/tensorflow/tensorflow/lite/nnapi/ |
| D | nnapi_handler.cc | 90 void NnApiHandler::SetDeviceName(const std::string& name) { in SetDeviceName() function in tflite::nnapi::NnApiHandler
97 NnApiHandler::SetDeviceName(name); in GetDeviceNameReturnsName()
103 NnApiHandler::SetDeviceName(name); in SetNnapiSupportedDevice()
|
| D | nnapi_handler.h | 312 static void SetDeviceName(const std::string& name);
|
| /external/tensorflow/tensorflow/c/experimental/op_handler/ |
| D | wrapper_operation.cc | 37 Status WrapperOperation::SetDeviceName(const char* name) { in SetDeviceName() function in tensorflow::WrapperOperation
38 return parent_op_->SetDeviceName(name); in SetDeviceName()
|
| D | wrapper_operation.h | 33 Status SetDeviceName(const char* name) override;
|
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_operation.cc | 50 Status TapeOperation::SetDeviceName(const char* name) { in SetDeviceName() function in tensorflow::gradients::TapeOperation
51 return parent_op_->SetDeviceName(name); in SetDeviceName()
|
| D | tape_operation.h | 30 Status SetDeviceName(const char* name) override;
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | abstract_operation.h | 80 virtual Status SetDeviceName(const char* name) = 0;
|
| D | c_api_unified_experimental_graph.cc | 128 Status SetDeviceName(const char* name) override { in SetDeviceName() function in tensorflow::tracing::graph::GraphOperation
|
| D | c_api.cc | 665 status->status = tensorflow::unwrap(op)->SetDeviceName(device_name); in TFE_OpSetDevice()
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 205 Status s = self->SetDeviceName(name); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| D | c_api_unified_experimental_mlir.cc | 146 Status SetDeviceName(const char* name) override;
450 Status MlirAbstractOp::SetDeviceName(const char* name) { in SetDeviceName() function in mlir::TF::__anon3f449e090111::MlirAbstractOp
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | partially_revived_objects.cc | 410 TF_RETURN_IF_ERROR(function_op->SetDeviceName(resource.device.c_str())); in CreateAllResourceHandles()
|