Home
last modified time | relevance | path

Searched refs:raw_val (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc420 const TensorProto& raw_val = array_size->attr().at("value").tensor(); in MaterializeShapes() local
421 if (raw_val.dtype() != DT_INT32) { in MaterializeShapes()
424 Tensor value(raw_val.dtype(), raw_val.tensor_shape()); in MaterializeShapes()
425 if (!value.FromProto(raw_val)) { in MaterializeShapes()
528 const TensorProto& raw_val = shape_node.attr().at("value").tensor(); in ExtractShape() local
529 if (raw_val.dtype() != DT_INT64 && raw_val.dtype() != DT_INT32) { in ExtractShape()
532 Tensor value(raw_val.dtype(), raw_val.tensor_shape()); in ExtractShape()
533 if (!value.FromProto(raw_val)) { in ExtractShape()
537 if (raw_val.dtype() == DT_INT64) { in ExtractShape()
1192 const TensorProto& raw_val = input_node->attr().at("value").tensor(); in EvaluateOneFoldable() local
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2200 const TensorProto& raw_val = fanin.node->attr().at("value").tensor(); in InferStatically() local
2201 *input_properties[i].mutable_value() = raw_val; in InferStatically()
2229 const TensorProto& raw_val = node.attr().at("value").tensor(); in InferStatically() local
2230 *output_properties[i].mutable_value() = raw_val; in InferStatically()