Searched refs:indices_array (Results 1 – 6 of 6) sorted by relevance
36 const Array& indices_array = model->GetArray(op->inputs[1]); in ResolveAttributes() local37 if (!indices_array.has_shape()) return false; in ResolveAttributes()41 if (indices_array.buffer->Length() == 0) return false; in ResolveAttributes()43 op->axis = indices_array.GetBuffer<ArrayDataType::kInt32>().data; in ResolveAttributes()
43 const auto& indices_array = model->GetArray(op->inputs[2]); in Run() local44 if (!indices_array.has_shape()) return ::tensorflow::OkStatus(); in Run()45 const auto& axis_data = indices_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
1394 const auto& indices_array = model->GetArray(op->inputs[1]); in ProcessGatherOperator() local1403 if (!input_array.has_shape() || !indices_array.has_shape()) { in ProcessGatherOperator()1414 const auto& indices_shape = indices_array.shape(); in ProcessGatherOperator()1435 const auto& indices_array = model->GetArray(op->inputs[1]); in ProcessGatherNdOperator() local1444 if (!input_array.has_shape() || !indices_array.has_shape()) { in ProcessGatherNdOperator()1449 const auto& indices_shape = indices_array.shape(); in ProcessGatherNdOperator()1957 const auto& indices_array = in ProcessOneHotOperator() local1959 if (!indices_array.has_shape()) { in ProcessOneHotOperator()1982 const int indices_dims = indices_array.shape().dimensions_count(); in ProcessOneHotOperator()1992 dim = indices_array.shape().dims(i); in ProcessOneHotOperator()[all …]
15 indices_array = np.array([[0, 0]])19 st = tf.SparseTensor(indices_array, value_array, dense_shape)
15 indices_array = np.array([[0, 0]])18 st = tf.SparseTensor(indices_array, value_array, dense_shape)
182 struct wl_array *indices_array in feedback_handle_tranche_formats() argument