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()
760 const auto& axis_array = model->GetArray(op->inputs[0]); in ProcessTensorFlowSplitOperator() local763 if (!axis_array.has_shape()) { in ProcessTensorFlowSplitOperator()767 CHECK(axis_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowSplitOperator()769 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitOperator()772 int axis = axis_array.GetBuffer<ArrayDataType::kInt32>().data[0]; in ProcessTensorFlowSplitOperator()831 const auto& axis_array = model->GetArray(op->inputs[2]); in ProcessTensorFlowSplitVOperator() local833 if (!axis_array.has_shape()) { in ProcessTensorFlowSplitVOperator()837 CHECK(axis_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowSplitVOperator()839 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitVOperator()842 int axis = axis_array.GetBuffer<ArrayDataType::kInt32>().data[0]; in ProcessTensorFlowSplitVOperator()[all …]
247 auto& axis_array = model->GetArray(unary_op->inputs[1]); in Run() local248 CHECK(axis_array.data_type == ArrayDataType::kInt32); in Run()255 sum_op->keep_dims, axis_array.GetBuffer<ArrayDataType::kInt32>().data, in Run()