|
|
|
@ -373,7 +373,7 @@ void GatherV2Kernel::DebugPrint(int64_t axis, const GeShape &x_shape, const GeSh
|
|
|
|
|
|
|
|
|
|
Status GatherV2Kernel::Compute(const OpDescPtr op_desc_ptr, const vector<ConstGeTensorPtr> &input,
|
|
|
|
|
vector<GeTensorPtr> &v_output) {
|
|
|
|
|
GELOGI("Enter GatherV2Kernel Process");
|
|
|
|
|
GELOGI("Enter GatherV2Kernel Process.");
|
|
|
|
|
Status ret = Check(op_desc_ptr, input, v_output);
|
|
|
|
|
if (ret != SUCCESS) {
|
|
|
|
|
GELOGW("param check failed");
|
|
|
|
@ -407,7 +407,8 @@ Status GatherV2Kernel::Compute(const OpDescPtr op_desc_ptr, const vector<ConstGe
|
|
|
|
|
// check input data type
|
|
|
|
|
auto x_data_type = tensor0->GetTensorDesc().GetDataType();
|
|
|
|
|
if (supported_type.find(x_data_type) == supported_type.end()) {
|
|
|
|
|
GELOGI("GatherV2Kernel does not support this Data type:%s.", TypeUtils::DataTypeToSerialString(x_data_type).c_str());
|
|
|
|
|
GELOGI("GatherV2Kernel does not support this Data type:%s.",
|
|
|
|
|
TypeUtils::DataTypeToSerialString(x_data_type).c_str());
|
|
|
|
|
return NOT_CHANGED;
|
|
|
|
|
}
|
|
|
|
|
// calc output shape
|
|
|
|
|