Home
last modified time | relevance | path

Searched refs:SetAttrTensor (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dgradients_internal.h56 Status SetAttrTensor(AbstractOperation*, const char* attr_name,
Dabstract_operation.h103 virtual Status SetAttrTensor(const char* attr_name,
Dc_api_unified_experimental_graph.cc223 Status SetAttrTensor(const char* attr_name, in SetAttrTensor() function in tensorflow::tracing::graph::GraphOperation
Dgradients.cc384 Status SetAttrTensor(AbstractOperation* op_, const char* attr_name, in SetAttrTensor() function
Dc_api.cc771 status->status = tensorflow::unwrap(op)->SetAttrTensor(attr_name, &interface); in TFE_OpSetAttrTensor()
/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.h47 Status SetAttrTensor(const char* attr_name,
Dtape_operation.cc118 Status TapeOperation::SetAttrTensor(const char* attr_name, in SetAttrTensor() function in tensorflow::gradients::TapeOperation
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.h109 Status SetAttrTensor(const char* attr_name,
Deager_operation.cc118 Status EagerOperation::SetAttrTensor(const char* attr_name, in SetAttrTensor() function in tensorflow::EagerOperation
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc166 Status SetAttrTensor(const char* attr_name,
482 Status MlirAbstractOp::SetAttrTensor(const char* attr_name, in SetAttrTensor() function in mlir::TF::__anon376bf20e0111::MlirAbstractOp
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc253 Status SetAttrTensor(const char* attr_name, in SetAttrTensor() function in tfrt::tf::__anond52f36620111::FakeOperation
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h529 tensorflow::Status SetAttrTensor(
Dc_api_tfrt.cc1758 tensorflow::Status OperationInterface::SetAttrTensor( in SetAttrTensor() function in tfrt::tf::OperationInterface
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_kernels.cc465 *status_ptr = eager_op->SetAttrTensor(entry.name, &interface); in PrepareAttributes()
756 handler, eager_op->SetAttrTensor(attr_name.c_str(), &interface)); in RuntimeFallbackKernel()