Searched refs:indices_array (Results 1 – 3 of 3) 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()
44 const auto& indices_array = model->GetArray(op->inputs[2]); in Run() local45 if (!indices_array.has_shape()) return ::tensorflow::Status::OK(); in Run()46 const auto& axis_data = indices_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
1374 const auto& indices_array = model->GetArray(op->inputs[1]); in ProcessGatherOperator() local1383 if (!input_array.has_shape() || !indices_array.has_shape()) { in ProcessGatherOperator()1394 const auto& indices_shape = indices_array.shape(); in ProcessGatherOperator()1415 const auto& indices_array = model->GetArray(op->inputs[1]); in ProcessGatherNdOperator() local1424 if (!input_array.has_shape() || !indices_array.has_shape()) { in ProcessGatherNdOperator()1429 const auto& indices_shape = indices_array.shape(); in ProcessGatherNdOperator()1933 const auto& indices_array = in ProcessOneHotOperator() local1935 if (!indices_array.has_shape()) { in ProcessOneHotOperator()1958 const int indices_dims = indices_array.shape().dimensions_count(); in ProcessOneHotOperator()1968 dim = indices_array.shape().dims(i); in ProcessOneHotOperator()[all …]