Home
last modified time | relevance | path

Searched refs:value_attr (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf.cc63 const auto& value_attr = node->attr().at("value"); in FilterPartitionedConstNodes() local
64 const tensorflow::TensorProto& tensor = value_attr.tensor(); in FilterPartitionedConstNodes()
83 const auto& value_attr = node->attr().at("value"); in InferFilterRank() local
84 const tensorflow::TensorProto& tensor = value_attr.tensor(); in InferFilterRank()
190 const auto& value_attr = const_node_parts[i]->attr().at("value"); in MaybeMergeConstNodes() local
191 const tensorflow::TensorProto& tensor = value_attr.tensor(); in MaybeMergeConstNodes()
218 const auto& value_attr = const_node_parts[i]->attr().at("value"); in MaybeMergeConstNodes() local
219 const tensorflow::TensorProto& tensor = value_attr.tensor(); in MaybeMergeConstNodes()
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow_test.cc163 AttrValue value_attr; in BuildConstNode() local
164 SetAttrValue(t, &value_attr); in BuildConstNode()
165 (*node->mutable_attr())["value"] = value_attr; in BuildConstNode()
498 AttrValue value_attr; in NodeWithTensorContent() local
499 SetAttrValue(t, &value_attr); in NodeWithTensorContent()
500 (*node->mutable_attr())["value"] = value_attr; in NodeWithTensorContent()
664 AttrValue value_attr; in TEST() local
665 SetAttrValue(2, &value_attr); in TEST()
666 (*node.mutable_attr())["Nsparse"] = value_attr; in TEST()
672 AttrValue value_attr; in TEST() local
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc612 const auto* value_attr = fanin_node_view->GetAttr(kAttrValue); in IsFaninPortDimsNIfConst() local
613 if (value_attr == nullptr) { in IsFaninPortDimsNIfConst()
617 if (!tensor.FromProto(value_attr->tensor())) { in IsFaninPortDimsNIfConst()
1514 const auto* value_attr = axis_node->GetAttr(kAttrValue); in IsReduceAxisSupported() local
1515 if (value_attr == nullptr) { in IsReduceAxisSupported()
1519 if (!tensor.FromProto(value_attr->tensor())) { in IsReduceAxisSupported()
2171 const auto* value_attr = regular_fanin_node->GetAttr(kAttrValue); in GetValueAttrFromConstInputNode() local
2172 if (value_attr == nullptr || value_attr->tensor().dtype() != DT_INT32) { in GetValueAttrFromConstInputNode()
2175 if (!tensor->FromProto(value_attr->tensor())) { in GetValueAttrFromConstInputNode()
Dgeneric_layout_optimizer_transposer_test.cc530 const auto* value_attr = const_perm_node->GetAttr("value"); in TEST_F() local
531 ASSERT_NE(value_attr, nullptr); in TEST_F()
534 ASSERT_TRUE(tensor.FromProto(value_attr->tensor())); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc895 auto value_attr = const_op.value().template cast<DenseElementsAttr>(); in matchAndRewrite() local
896 if (value_attr.getType().getRank() != 0) return failure(); in matchAndRewrite()
897 ReplaceConstOp(loc, const_op, value_attr, rewriter); in matchAndRewrite()
903 DenseElementsAttr value_attr, in ReplaceConstOp() argument
905 Value std_tensor_const = rewriter.create<mlir::ConstantOp>(loc, value_attr); in ReplaceConstOp()
909 DenseElementsAttr value_attr, in ReplaceConstOp() argument
912 rewriter.create<mlir::ConstantOp>(loc, value_attr.getValue({})); in ReplaceConstOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc562 TF_ASSIGN_OR_RETURN(auto value_attr, in RewriteWithArgs()
566 mlir::UnknownLoc::get(module_op.getContext()), value_attr); in RewriteWithArgs()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc257 auto value_attr = const_op.value().dyn_cast<DenseIntElementsAttr>(); in fold() local
258 if (!value_attr || value_attr.getNumElements() != 1) return None; in fold()
260 auto value_ty = value_attr.getType(); in fold()
263 auto splat = value_attr.getSplatValue<IntegerAttr>(); in fold()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc141 tensorflow::Safe_PyObjectPtr value_attr( in InputTFE_InputTensorHandles() local
143 if (value_attr) { in InputTFE_InputTensorHandles()
145 (input_tensor_handles)[i] = EagerTensor_Handle(value_attr.get()); in InputTFE_InputTensorHandles()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3219 TF_ASSIGN_OR_RETURN(auto value_attr, ConvertTensor(value, &builder)); in CreateSavedModelIR()
3228 value_attr, in CreateSavedModelIR()
3243 TF_ASSIGN_OR_RETURN(auto value_attr, in CreateSavedModelIR()
3248 value_attr, in CreateSavedModelIR()
3249 /*type=*/mlir::TypeAttr::get(value_attr.Attribute::getType()), in CreateSavedModelIR()