Searched refs:crops_array (Results 1 – 2 of 2) sorted by relevance
49 const auto& crops_array = model->GetArray(op->inputs[2]); in Run() local50 if (!crops_array.has_shape()) return ::tensorflow::Status::OK(); in Run()51 const std::vector<int>& crops_dims = crops_array.shape().dims(); in Run()58 crops_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
1334 const auto& crops_array = model->GetArray(op->inputs[2]); in ProcessBatchToSpaceNDOperator() local1336 const auto& crops_array_shape = crops_array.shape(); in ProcessBatchToSpaceNDOperator()1353 if (!crops_array.buffer) { in ProcessBatchToSpaceNDOperator()1356 QCHECK(crops_array.data_type == ArrayDataType::kInt32); in ProcessBatchToSpaceNDOperator()1357 const auto& crops_data = crops_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessBatchToSpaceNDOperator()