Searched refs:axis_array (Results 1 – 4 of 4) sorted by relevance
47 const auto& axis_array = model->GetArray(expand_op->inputs[1]); in Run() local48 if (!axis_array.has_shape()) { in Run()52 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1); in Run()53 if (!axis_array.buffer) { in Run()57 int axis = axis_array.GetBuffer<ArrayDataType::kInt32>().data[0]; in Run()
56 const auto& axis_array = model->GetArray(axis_name); in Run() local57 if (!axis_array.buffer) { in Run()63 CHECK(axis_array.data_type == ArrayDataType::kInt32); in Run()64 const auto& axis_data = axis_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
773 const auto& axis_array = model->GetArray(op->inputs[0]); in ProcessTensorFlowSplitOperator() local776 if (!axis_array.has_shape()) { in ProcessTensorFlowSplitOperator()780 CHECK(axis_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowSplitOperator()782 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitOperator()785 int axis = axis_array.GetBuffer<ArrayDataType::kInt32>().data[0]; in ProcessTensorFlowSplitOperator()844 const auto& axis_array = model->GetArray(op->inputs[2]); in ProcessTensorFlowSplitVOperator() local846 if (!axis_array.has_shape()) { in ProcessTensorFlowSplitVOperator()850 CHECK(axis_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowSplitVOperator()852 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitVOperator()855 int axis = axis_array.GetBuffer<ArrayDataType::kInt32>().data[0]; in ProcessTensorFlowSplitVOperator()[all …]
253 auto& axis_array = model->GetArray(unary_op->inputs[1]); in Run() local254 CHECK(axis_array.data_type == ArrayDataType::kInt32); in Run()261 sum_op->keep_dims, axis_array.GetBuffer<ArrayDataType::kInt32>().data, in Run()