Home
last modified time | relevance | path

Searched refs:GetNodeTensorAttr (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather_test.cc216 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 …]
Dfold_old_batch_norms.cc71 Tensor mean = GetNodeTensorAttr(mean_node, "value"); in GetScaleAndOffsetValues()
72 Tensor variance = GetNodeTensorAttr(variance_node, "value"); in GetScaleAndOffsetValues()
73 Tensor beta = GetNodeTensorAttr(beta_node, "value"); in GetScaleAndOffsetValues()
74 Tensor gamma = GetNodeTensorAttr(gamma_node, "value"); in GetScaleAndOffsetValues()
117 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FuseScaleOffsetToConvWeights()
238 Tensor axis = GetNodeTensorAttr(axis_node, "value"); in FuseBatchNormWithConvConcat()
249 Tensor weights0 = GetNodeTensorAttr(weights0_node, "value"); in FuseBatchNormWithConvConcat()
Dfuse_quantized_convolution.cc171 GetNodeTensorAttr(*min_input_node, "value").flat<float>()(0); in FuseQuantizedConvolutionAndRequantize()
173 GetNodeTensorAttr(*max_input_node, "value").flat<float>()(0); in FuseQuantizedConvolutionAndRequantize()
175 GetNodeTensorAttr(*min_filter_node, "value").flat<float>()(0); in FuseQuantizedConvolutionAndRequantize()
177 GetNodeTensorAttr(*max_filter_node, "value").flat<float>()(0); in FuseQuantizedConvolutionAndRequantize()
Dfold_batch_norms.cc70 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FoldBatchNorms()
71 Tensor mul_values = GetNodeTensorAttr(mul_values_node, "value"); in FoldBatchNorms()
Dflatten_atrous.cc63 GetNodeTensorAttr(space_to_batch_block_shape_node, "value"); in FlattenAtrousConv()
68 const Tensor& filter = GetNodeTensorAttr(filter_node, "value"); in FlattenAtrousConv()
Dquantize_weights_test.cc146 return GetNodeTensorAttr(*node_def, "value"); in TEST_F()
Dtransform_utils.h104 Tensor GetNodeTensorAttr(const NodeDef& node, const string& key);
Dsparsify_gather.cc338 weight = GetNodeTensorAttr(weights_node, "value"); in SparsifyGatherInternal()
Dtransform_utils.cc151 Tensor GetNodeTensorAttr(const NodeDef& node, const string& key) { in GetNodeTensorAttr() function
Dtransform_utils_test.cc215 Tensor result = GetNodeTensorAttr(node, "foo"); in TestGetNodeTensorAttr()