| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.cc | 43 TF_RETURN_IF_ERROR(varhandle_op->SetAttrType("dtype", dtype)); in CreateUninitializedResourceVariable()
75 TF_RETURN_IF_ERROR(assign_op->SetAttrType("dtype", dtype)); in AssignVariable()
89 TF_RETURN_IF_ERROR(read_op->SetAttrType("dtype", dtype)); in ReadVariable()
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | resource_variable_ops.cc | 47 TF_RETURN_IF_ERROR(op_ptr->SetAttrType("dtype", dtype)); in VarHandleOp()
73 TF_RETURN_IF_ERROR(op_ptr->SetAttrType("dtype", dtype)); in ReadVariableOp()
|
| D | array_ops.cc | 114 TF_RETURN_IF_ERROR(op_ptr->SetAttrType("out_type", out_type)); in Shape()
|
| /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 | 94 virtual Status SetAttrType(const char* attr_name, DataType value) = 0;
|
| D | c_api_unified_experimental_graph.cc | 192 Status SetAttrType(const char* const attr_name, DataType value) override { in SetAttrType() function in tensorflow::tracing::graph::GraphOperation
336 TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype)); in AddParameter()
|
| D | gradients.cc | 345 Status SetAttrType(AbstractOperation* op_, const char* attr_name, in SetAttrType() function
348 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 | 736 auto s = tensorflow::unwrap(op)->SetAttrType( in TFE_OpSetAttrType()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | op_cache_test.cc | 62 ASSERT_TRUE(op_interface_->SetAttrType("T", tensorflow::DT_INT8).ok()); in SetUp()
|
| D | c_api_tfrt.h | 520 tensorflow::Status SetAttrType(const char* attr_name,
|
| D | c_api_tfrt.cc | 1649 tensorflow::Status OperationInterface::SetAttrType(const char* attr_name, in SetAttrType() function in tfrt::tf::OperationInterface
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 217 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
|
| D | execute.cc | 747 wrapped_op->SetAttrType(name_type.first.data(), name_type.second)); in AddMixedTypeListAttrs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| D | c_api_unified_experimental_mlir.cc | 158 Status SetAttrType(const char* attr_name,
303 Status MlirAbstractOp::SetAttrType(const char* attr_name, in SetAttrType() function in mlir::TF::__anon376bf20e0111::MlirAbstractOp
|
| /external/tensorflow/tensorflow/c/experimental/ops/gen/cpp/golden/ |
| D | testing_ops.cc.golden | 117 TF_RETURN_IF_ERROR(op_ptr->SetAttrType("dtype", dtype));
|
| /external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| D | op_handler_selector_test.cc | 237 Status SetAttrType(const char* attr_name, in SetAttrType() function in tfrt::tf::__anond52f36620111::FakeOperation
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 448 *status_ptr = eager_op->SetAttrType(entry.name, tf_dtype); in PrepareAttributes()
762 handler, eager_op->SetAttrType(attr_name.c_str(), dtype)); in RuntimeFallbackKernel()
|