| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.cc | 45 TF_RETURN_IF_ERROR(varhandle_op->SetAttrType("dtype", dtype)); in CreateUninitializedResourceVariable()
76 TF_RETURN_IF_ERROR(assign_op->SetAttrType("dtype", dtype)); in AssignVariable()
90 TF_RETURN_IF_ERROR(read_op->SetAttrType("dtype", dtype)); in ReadVariable()
|
| /external/tensorflow/tensorflow/c/experimental/op_handler/ |
| D | wrapper_operation.cc | 60 Status WrapperOperation::SetAttrType(const char* attr_name, DataType value) { in SetAttrType() function in tensorflow::WrapperOperation
61 return parent_op_->SetAttrType(attr_name, value); in SetAttrType()
|
| D | wrapper_operation.h | 43 Status SetAttrType(const char* attr_name, DataType value) override;
|
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_operation.cc | 83 Status TapeOperation::SetAttrType(const char* attr_name, DataType value) { in SetAttrType() function in tensorflow::gradients::TapeOperation
85 return parent_op_->SetAttrType(attr_name, value); in SetAttrType()
|
| D | tape_operation.h | 40 Status SetAttrType(const char* attr_name, DataType value) override;
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients_internal.h | 48 Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value,
|
| D | abstract_operation.h | 93 virtual Status SetAttrType(const char* attr_name, DataType value) = 0;
|
| D | c_api_unified_experimental_graph.cc | 193 Status SetAttrType(const char* const attr_name, DataType value) override { in SetAttrType() function in tensorflow::tracing::graph::GraphOperation
365 TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype)); in AddParameter()
|
| D | gradients.cc | 344 Status SetAttrType(AbstractOperation* op_, const char* attr_name, in SetAttrType() function
347 return op_->SetAttrType(attr_name, value); in SetAttrType()
|
| D | c_api_unified_experimental.cc | 215 unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value)); in TF_AbstractOpSetAttrType()
|
| D | c_api.cc | 755 auto s = tensorflow::unwrap(op)->SetAttrType( in TFE_OpSetAttrType()
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 215 Status s = self->SetAttrType(attr_name, value); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation.h | 102 Status SetAttrType(const char* attr_name, DataType value) override;
|
| D | eager_operation.cc | 70 Status EagerOperation::SetAttrType(const char* attr_name, DataType value) { in SetAttrType() function in tensorflow::EagerOperation
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| D | c_api_unified_experimental_mlir.cc | 158 Status SetAttrType(const char* attr_name,
302 Status MlirAbstractOp::SetAttrType(const char* attr_name, in SetAttrType() function in mlir::TF::__anon3f449e090111::MlirAbstractOp
|