Searched refs:GetIntAttr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 90 int64 GetIntAttr(const NodeDef& node, const std::string& attr_name) { in GetIntAttr() function
617 int num_outputs = GetIntAttr(node, multiples); in GetOutputNamesFromNodeDef()
665 int num_outputs = GetIntAttr(node, multiples); in GetOutputTypesFromNodeDef()
1005 op->block_size = GetIntAttr(node, "block_size"); in ConvertDepthToSpaceOperator()
1028 op->block_size = GetIntAttr(node, "block_size"); in ConvertSpaceToDepthOperator()
1062 op->seed = GetIntAttr(node, "seed"); in ConvertRandomUniform()
1063 op->seed2 = GetIntAttr(node, "seed2"); in ConvertRandomUniform()
1123 op->num_bits = HasAttr(node, "num_bits") ? GetIntAttr(node, "num_bits") : 8; in ConvertFakeQuantWithMinMaxArgs()
1145 op->num_bits = HasAttr(node, "num_bits") ? GetIntAttr(node, "num_bits") : 8; in ConvertFakeQuantWithMinMaxVars()
1182 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitOperator()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op_test.cc | 43 int GetIntAttr(const string& name) { in GetIntAttr() function in tensorflow::LRNFloatTest
63 const int64 depth_radius = GetIntAttr("depth_radius"); in Compare()
|