Searched refs:axis_tensor (Results 1 – 12 of 12) sorted by relevance
107 auto axis_tensor = input_args[1]->cast<abstract::AbstractTensorPtr>(); in InferShape() local108 auto axis_shape = axis_tensor->shape()->shape(); in InferShape()144 auto axis_tensor = axis_ptr->cast<tensor::TensorPtr>(); in InferShape() local145 MS_EXCEPTION_IF_NULL(axis_tensor); in InferShape()146 size_t data_size = LongToSize(axis_tensor->DataSize()); in InferShape()149 auto shape_data = reinterpret_cast<int *>(axis_tensor->data_c()); in InferShape()156 auto shape_data2 = reinterpret_cast<int64_t *>(axis_tensor->data_c()); in InferShape()
56 auto axis_tensor = axis_value_ptr->cast<tensor::TensorPtr>(); in GatherInfer() local57 MS_EXCEPTION_IF_NULL(axis_tensor); in GatherInfer()58 axis_val = *static_cast<int64_t *>(axis_tensor->data_c()); in GatherInfer()
37 auto axis_tensor = AttrToTensor(&axis_attr, 1, false, kNumberTypeInt32, tensor_bufs); in TransferGatherAttr() local38 if (axis_tensor == nullptr) { in TransferGatherAttr()42 dst_tensors->push_back(axis_tensor); in TransferGatherAttr()
131 mindspore::MSTensor axis_tensor = this->in_tensors_[1]; in GetAxis() local132 if (axis_tensor.Data() == nullptr) { in GetAxis()136 if (axis_tensor.DataType() != DataType::kNumberTypeInt32) { in GetAxis()139 int *axis_data = reinterpret_cast<int *>(axis_tensor.MutableData()); in GetAxis()140 for (int i = 0; i < axis_tensor.ElementNum(); i++) { in GetAxis()
38 const auto &axis_tensor = tflite_subgraph->tensors.at(tflite_op->inputs[1]); in Parse() local39 MS_CHECK_TRUE_MSG(axis_tensor != nullptr, nullptr, "axis_tensor is nullptr"); in Parse()40 const auto &buf_data = tflite_model->buffers.at(axis_tensor->buffer); in Parse()
56 const auto &axis_tensor = tflite_subgraph->tensors.at(tflite_op->inputs.at(2)); in Parse() local57 if (axis_tensor == nullptr) { in Parse()61 auto &axis_buf_data = tflite_model->buffers.at(axis_tensor->buffer); in Parse()
48 const auto &axis_tensor = tflite_subgraph->tensors.at(tflite_op->inputs.at(0)); in Parse() local49 if (axis_tensor == nullptr) { in Parse()53 auto &axis_buf_data = tflite_model->buffers.at(axis_tensor->buffer); in Parse()
38 auto &axis_tensor = attr_value.tensor(); in Parse() local39 prim->set_axis(axis_tensor.int_val(0)); in Parse()
43 auto &axis_tensor = attr_value.tensor(); in Parse() local44 prim->set_axis(axis_tensor.int_val(0)); in Parse()
63 auto axis_tensor = in_tensors_.at(1); in ReSize() local64 CHECK_NULL_RETURN(axis_tensor); in ReSize()65 int *axis_data = reinterpret_cast<int *>(axis_tensor->data()); in ReSize()
596 auto axis_tensor = axis_value_ptr->cast<tensor::TensorPtr>(); in InferImplGatherV2() local597 MS_EXCEPTION_IF_NULL(axis_tensor); in InferImplGatherV2()598 axis_val = *static_cast<int64_t *>(axis_tensor->data_c()); in InferImplGatherV2()