Searched refs:GetNodeTensorAttr (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 216 GetNodeTensorAttr(*(node_lookup.at("w/part_1/indices")), "value")); in TestSinglePartition()
224 GetNodeTensorAttr(*(node_lookup.at("w/part_1/values")), "value"), 1e-5); in TestSinglePartition()
244 GetNodeTensorAttr(*(node_lookup.at("gather/Const")), "value"), 1e-5); in TestSinglePartition()
252 GetNodeTensorAttr(*(node_lookup.at("gather/ExpandDims/Const")), in TestSinglePartition()
438 GetNodeTensorAttr(*(node_lookup.at("w1/part_1/indices")), "value")); in TestMultiPartition()
446 GetNodeTensorAttr(*(node_lookup.at("w1/part_1/values")), "value"), in TestMultiPartition()
467 GetNodeTensorAttr(*(node_lookup.at("gather1/Const")), "value"), 1e-5); in TestMultiPartition()
475 GetNodeTensorAttr(*(node_lookup.at("gather1/ExpandDims/Const")), in TestMultiPartition()
487 GetNodeTensorAttr(*(node_lookup.at("w2/part_1/indices")), "value")); in TestMultiPartition()
495 GetNodeTensorAttr(*(node_lookup.at("w2/part_1/values")), "value"), in TestMultiPartition()
[all …]
|
| D | fold_old_batch_norms.cc | 70 Tensor mean = GetNodeTensorAttr(mean_node, "value"); in GetScaleAndOffsetValues()
71 Tensor variance = GetNodeTensorAttr(variance_node, "value"); in GetScaleAndOffsetValues()
72 Tensor beta = GetNodeTensorAttr(beta_node, "value"); in GetScaleAndOffsetValues()
73 Tensor gamma = GetNodeTensorAttr(gamma_node, "value"); in GetScaleAndOffsetValues()
116 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FuseScaleOffsetToConvWeights()
237 Tensor axis = GetNodeTensorAttr(axis_node, "value"); in FuseBatchNormWithConvConcat()
248 Tensor weights0 = GetNodeTensorAttr(weights0_node, "value"); in FuseBatchNormWithConvConcat()
|
| D | fold_batch_norms.cc | 69 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FoldBatchNorms()
70 Tensor mul_values = GetNodeTensorAttr(mul_values_node, "value"); in FoldBatchNorms()
|
| D | flatten_atrous.cc | 63 GetNodeTensorAttr(space_to_batch_block_shape_node, "value"); in FlattenAtrousConv()
68 const Tensor& filter = GetNodeTensorAttr(filter_node, "value"); in FlattenAtrousConv()
|
| D | quantize_weights_test.cc | 146 return GetNodeTensorAttr(*node_def, "value"); in TEST_F()
|
| D | transform_utils.h | 104 Tensor GetNodeTensorAttr(const NodeDef& node, const string& key);
|
| D | sparsify_gather.cc | 337 weight = GetNodeTensorAttr(weights_node, "value"); in SparsifyGatherInternal()
|
| D | transform_utils.cc | 153 Tensor GetNodeTensorAttr(const NodeDef& node, const string& key) { in GetNodeTensorAttr() function
|
| D | transform_utils_test.cc | 215 Tensor result = GetNodeTensorAttr(node, "foo"); in TestGetNodeTensorAttr()
|