Searched refs:tensor_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lift_variables.cc | 137 ElementsAttr tensor_attr = tensor_attr_or.ValueOrDie(); in LiftVariablesFromSession() local
141 builder.getStringAttr(name), tensor_attr, in LiftVariablesFromSession()
142 TypeAttr::get(tensor_attr.getType()), builder.getUnitAttr()); in LiftVariablesFromSession()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2025 TensorProto* tensor_attr = in TestConvertConst() local
2027 tensor_attr->Clear(); in TestConvertConst()
2030 tensor.AsProtoTensorContent(tensor_attr); in TestConvertConst()
2032 tensor.shape().AsProto(tensor_attr->mutable_tensor_shape()); in TestConvertConst()
2033 tensor_attr->set_dtype(tensor.dtype()); in TestConvertConst()
2036 CopyTensorElements<float>(tensor, tensor_attr->mutable_float_val()); in TestConvertConst()
2038 CopyTensorElements<int32>(tensor, tensor_attr->mutable_int_val()); in TestConvertConst()
2040 tensor.AsProtoField(tensor_attr); in TestConvertConst()
2107 TensorProto* tensor_attr = in TEST_F() local
2109 tensor_attr->Clear(); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 771 mlir::Attribute tensor_attr = inst->getAttr("value"); in BuildTensor() local
773 tensor_attr.getType().cast<TensorType>().getShape(); in BuildTensor()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2019 AttrValue tensor_attr; in ReplaceOperationWithConstant() local
2020 Status s = CreateConstantTensorAttrValue(dtype, value, shape, &tensor_attr); in ReplaceOperationWithConstant()
2028 return ReplaceOperationWithConstantTensor(dtype, tensor_attr.mutable_tensor(), in ReplaceOperationWithConstant()
|