Searched defs:tensor_attr (Results 1 – 8 of 8) sorted by relevance
44 ElementsAttr tensor_attr = tensor_attr_or.ValueOrDie(); in InitializeVariable() local
137 ElementsAttr tensor_attr = tensor_attr_or.ValueOrDie(); in LiftVariablesFromSession() local
402 if (auto tensor_attr = attr.dyn_cast<mlir::TF::TensorProtoAttr>()) in ConvertToTensorProto() local
401 if (auto tensor_attr = attr.dyn_cast<mlir::tf_type::TensorProtoAttr>()) in ConvertToTensorProto() local
193 auto tensor_attr = "tensor_attr"; in TEST() local
905 auto tensor_attr = inst->getAttr("value").cast<mlir::TypedAttr>(); in BuildTensor() local
2068 AttrValue tensor_attr; in ReplaceOperationWithConstant() local
2410 TensorProto* tensor_attr = in TestConvertConst() local2497 TensorProto* tensor_attr = in TEST_F() local