Home
last modified time | relevance | path

Searched defs: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
83 const auto& value_attr = node->attr().at("value"); in InferFilterRank() local
190 const auto& value_attr = const_node_parts[i]->attr().at("value"); in MaybeMergeConstNodes() local
218 const auto& value_attr = const_node_parts[i]->attr().at("value"); in MaybeMergeConstNodes() local
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow_test.cc163 AttrValue value_attr; in BuildConstNode() local
498 AttrValue value_attr; in NodeWithTensorContent() local
664 AttrValue value_attr; in TEST() local
672 AttrValue value_attr; in TEST() local
683 AttrValue value_attr; in TEST() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt_data.cc55 T ConstAttrToTypeAttr(ElementsAttr value_attr) { in ConstAttrToTypeAttr()
/external/tensorflow/tensorflow/core/ir/
Dops.cc200 if (StringAttr value_attr = op->getAttrOfType<StringAttr>(keyword)) in printOperation() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc612 const auto* value_attr = fanin_node_view->GetAttr(kAttrValue); in IsFaninPortDimsNIfConst() local
1514 const auto* value_attr = axis_node->GetAttr(kAttrValue); in IsReduceAxisSupported() local
2172 const auto* value_attr = regular_fanin_node->GetAttr(kAttrValue); in GetValueAttrFromConstInputNode() local
Dgeneric_layout_optimizer_transposer_test.cc530 const auto* value_attr = const_perm_node->GetAttr("value"); in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc1386 auto value_attr = const_op.value().cast<DenseElementsAttr>(); in matchAndRewrite() local
1404 auto value_attr = const_op.value().cast<DenseElementsAttr>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc143 tensorflow::Safe_PyObjectPtr value_attr( in InputTFE_InputTensorHandles() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc258 auto value_attr = const_op.value().dyn_cast<DenseIntElementsAttr>(); in fold() local