Home
last modified time | relevance | path

Searched refs:op_name_ (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/eager/
Dop_cache.h83 op_name_(op_name), in CacheKey()
89 op_name_(other.op_name_), in CacheKey()
96 op_name_ = op_name_concrete_.data(); in CacheKey()
113 op_name_concrete_ = op_name_.str(); in MakeConcrete()
135 return (lhs.op_name_ == rhs.op_name_ &&
139 string_view OpName() { return op_name_; } in OpName()
149 string_view op_name_, device_name_; variable
Dc_api_tfrt.cc1287 op_name_ = op; in Reset()
1296 tensorflow::Status s = tensorflow::OpDefForOp(op_name_, &op_def_); in Reset()
1445 op_name_, op_handler, device_name_, dtypes, this); in Initialize()
1448 StrCat("Cannot obtain CoreRuntimeOp: ", op_name_, in Initialize()
1459 GetFuncAttr(attrs_, op_name_, *context_->GetEagerContext()->FuncLibDef(), in Initialize()
1466 context_->GetOpCache().GetOrAddXlaOp(op_name_, context_); in Initialize()
1469 StrCat("Cannot initialize xla function ", op_name_, in Initialize()
1541 op_name_, device_name_, dev_set, context_->GetEagerContext(), corert, in Initialize()
1552 TF_RETURN_IF_ERROR(context_->RunMetadataRecordFunction(op_name_)); in Initialize()
Dc_api_tfrt.h491 const std::string& Name() const override { return op_name_; } in Name()
597 std::string op_name_; variable
/external/tensorflow/tensorflow/cc/framework/
Dscope.cc92 op_name_(""), in Impl()
110 op_name_(op_name), in Impl()
134 op_name_(other.impl()->op_name_), in Impl()
151 op_name_(other.impl()->op_name_), in Impl()
169 op_name_(op_name), in Impl()
186 op_name_(other.impl()->op_name_), in Impl()
204 op_name_(other.impl()->op_name_), in Impl()
222 op_name_(other.impl()->op_name_), in Impl()
243 op_name_(other.impl()->op_name_), in Impl()
261 op_name_(other.impl()->op_name_), in Impl()
[all …]
Dscope_internal.h112 const string op_name_ = ""; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dunimplemented_operation_parser.h33 : op_name_(op_name) {} in UnimplementedOperationParser()
38 return absl::UnimplementedError(op_name_); in IsSupported()
44 return absl::UnimplementedError(op_name_); in Parse()
48 std::string op_name_;
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.h104 op_name_ = op; in Reset()
113 const string& op_name() const { return op_name_; } in op_name()
145 "' found in AttrBuilder for ", op_name_); in Get()
203 string op_name_; // Conceptually const, but can't be because of Reset(...) variable
Dattr_builder.cc144 "' found in AttrBuilder for ", op_name_); \
166 "' found in AttrBuilder for ", op_name_); in Get()
323 node_def_.set_name(op_name_); in InitializeNodeDef()
324 node_def_.set_op(op_name_); in InitializeNodeDef()
Deager_operation.h217 const char* op_name() const { return op_name_; } in op_name()
244 const char* op_name_ = nullptr; variable
Deager_operation.cc350 op_name_ = op; in Reset()
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc177 op_name_ = op; in Reset()
183 const std::string& Name() const override { return op_name_; } in Name()
317 std::string op_name_; member in tfrt::tf::__anond52f36620111::FakeOperation
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtfrt_op_kernel.cc203 TFRTOpMetaBuilder::TFRTOpMetaBuilder(StringPiece op_name) : op_name_(op_name) {} in TFRTOpMetaBuilder()
233 const string& TFRTOpMetaBuilder::op_name() const { return op_name_; } in op_name()
Dtfrt_op_kernel.h212 string op_name_;
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen.cc139 op_name_ = function_name_; in GenEagerPythonOp()
140 absl::ConsumePrefix(&op_name_, "_"); in GenEagerPythonOp()
207 StringPiece op_name_; member in tensorflow::__anon20a011d50111::GenEagerPythonOp
639 return strings::StrCat("raise RuntimeError(\"", op_name_, in GetEagerNotAllowedError()
649 "' argument to \"\n", indentation, " \"'", op_name_, in ExpectListArg()
677 op_name_, "' Op with length %d \"\n", indentation, in GetEagerFunctionSetup()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc85 std::string op_name_; member in tflite::gpu::ThinPointwiseFuser
393 op_name_ += "dw_conv"; in AddDepthwiseConvNode()
503 op_name_ += "->conv1x1"; in AddConvNode()
596 gpu_subgraph->operations[0].name = op_name_; in Finalize()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc211 const char* op_name_ = nullptr; member in mlir::TF::__anon376bf20e0111::MlirAbstractOp
317 if (op_name_) { in SetOpName()
320 op_name_ = op_name; in SetOpName()
/external/tensorflow/tensorflow/core/kernels/
Ddebug_ops.h415 OP_REQUIRES_OK(context, context->GetAttr("op_name", &op_name_)); in DebugIdentityV2Op()
438 tfdbg_context_id_, device_name_, op_name_, in Compute()
448 string op_name_; variable
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_graph.cc310 const char* op_name_ = nullptr; member in tensorflow::tracing::graph::GraphOperation