Home
last modified time | relevance | path

Searched defs:tensor_attr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables.cc136 ElementsAttr tensor_attr = tensor_attr_or.ValueOrDie(); in LiftVariablesFromSession() local
Dinitialize_variables_in_session_init.cc68 ElementsAttr tensor_attr = tensor_attr_or.ValueOrDie(); in InitializeVariable() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc825 mlir::Attribute tensor_attr = inst->getAttr("value"); in BuildTensor() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2035 AttrValue tensor_attr; in ReplaceOperationWithConstant() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1947 TensorProto* tensor_attr = in TestConvertConst() local
2032 TensorProto* tensor_attr = in TEST_F() local