|
|
@@ -96,6 +96,7 @@ Status RsqrtKernel::RsqrtCompute(ConstGeTensorPtr &input_tensor_ptr, GeTensorPtr |
|
|
|
default: |
|
|
|
GELOGW("Input data type must be FP16, FP32 and DOUBLE."); |
|
|
|
return NOT_CHANGED; |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
GE_IF_BOOL_EXEC(output_tensor_ptr->SetData(reinterpret_cast<uint8_t *>(buf.get()), data_size) != GRAPH_SUCCESS, |
|
|
@@ -136,6 +137,7 @@ Status RsqrtKernel::Compute(const OpDescPtr op_desc_ptr, const std::vector<Const |
|
|
|
default: |
|
|
|
GELOGW("Input data type must be FP16, FP32 and DOUBLE."); |
|
|
|
return NOT_CHANGED; |
|
|
|
break; |
|
|
|
} |
|
|
|
if (ret != SUCCESS) { |
|
|
|
GELOGW("Rsqrt folding failed."); |
|
|
|