/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
D | resolve_svdf.cc | 63 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/ |
D | import_tensorflow_test.cc | 163 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/ |
D | generic_layout_optimizer_transposer.cc | 612 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()
|
D | generic_layout_optimizer_transposer_test.cc | 530 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/ |
D | legalize_to_linalg.cc | 895 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/ |
D | compile_mlir_util.cc | 562 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/ |
D | tf_ops_n_z.cc | 257 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/ |
D | tfe_wrapper.cc | 141 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/ |
D | import_model.cc | 3219 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()
|