Home
last modified time | relevance | path

Searched refs:value_split (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc729 std::vector<absl::string_view> value_split = in RuntimeFallbackKernel() local
733 if (value_split[0] == "string") { in RuntimeFallbackKernel()
736 eager_op->SetAttrString(attr_name.c_str(), value_split[1].data(), in RuntimeFallbackKernel()
737 value_split[1].size())); in RuntimeFallbackKernel()
738 } else if (value_split[0] == "bool") { in RuntimeFallbackKernel()
741 handler, ParseBoolAttrValue(value_split[1], &bool_val)); in RuntimeFallbackKernel()
744 } else if (value_split[0] == "int") { in RuntimeFallbackKernel()
747 handler, ParseIntAttrValue(value_split[1], &int_val)); in RuntimeFallbackKernel()
750 } else if (value_split[0] == "tftensor") { in RuntimeFallbackKernel()
753 ParseTensorAttrValue(value_split[1], &t)); in RuntimeFallbackKernel()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dattr_util.cc108 std::vector<absl::string_view> value_split = tfd::AttrValueSplit(attr_value); in AddOpAttr() local
109 auto& type = value_split[0]; in AddOpAttr()
110 auto& value = value_split[1]; in AddOpAttr()