Home
last modified time | relevance | path

Searched refs:GetIntAttr (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_data.i499 if (!GetIntAttr(item, "edge_padding_low", &edge_padding_low)
500 || !GetIntAttr(item, "edge_padding_high", &edge_padding_high)
501 || !GetIntAttr(item, "interior_padding", &interior_padding)) {
525 if (!GetIntAttr($input, "input_batch_dimension", &value)) {
530 if (!GetIntAttr($input, "input_feature_dimension", &value)) {
535 if (!GetIntAttr($input, "output_batch_dimension", &value)) {
540 if (!GetIntAttr($input, "output_feature_dimension", &value)) {
545 if (!GetIntAttr($input, "kernel_output_feature_dimension", &value)) {
550 if (!GetIntAttr($input, "kernel_input_feature_dimension", &value)) {
595 if (!GetIntAttr($input, "index_vector_dim", &value)) {
[all …]
Dnumpy_bridge.h141 bool GetIntAttr(PyObject* o, const char* field, int64* result);
Dlocal_computation_builder.i246 if (!GetIntAttr($input, "num_replicas", &num_replicas)) {
Dnumpy_bridge.cc570 bool GetIntAttr(PyObject* o, const char* field, int64* result) { in GetIntAttr() function
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc88 int64 GetIntAttr(const NodeDef& node, const string& attr_name) { in GetIntAttr() function
773 op->block_size = GetIntAttr(node, "block_size"); in ConvertDepthToSpaceOperator()
796 op->block_size = GetIntAttr(node, "block_size"); in ConvertSpaceToDepthOperator()
830 op->seed = GetIntAttr(node, "seed"); in ConvertRandomUniform()
831 op->seed2 = GetIntAttr(node, "seed2"); in ConvertRandomUniform()
872 op->num_bits = HasAttr(node, "num_bits") ? GetIntAttr(node, "num_bits") : 8; in ConvertFakeQuantWithMinMaxArgs()
894 op->num_bits = HasAttr(node, "num_bits") ? GetIntAttr(node, "num_bits") : 8; in ConvertFakeQuantWithMinMaxVars()
931 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitOperator()
950 const int num_split = GetIntAttr(node, "num_split"); in ConvertSplitVOperator()
1000 lrn->range = GetIntAttr(node, "depth_radius"); in ConvertLRNOperator()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dlrn_op_test.cc40 int GetIntAttr(const string& name) { in GetIntAttr() function in tensorflow::LRNFloatTest
60 const int64 depth_radius = GetIntAttr("depth_radius"); in Compare()