Searched refs:block_shape_array (Results 1 – 4 of 4) sorted by relevance
65 const auto& block_shape_array = model->GetArray(op->inputs[1]); in Run() local66 if (!block_shape_array.has_shape()) return ::tensorflow::Status::OK(); in Run()67 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims(); in Run()70 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
68 const auto& block_shape_array = in Run() local70 if (!block_shape_array.has_shape()) return ::tensorflow::Status::OK(); in Run()71 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims(); in Run()74 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
190 const auto& block_shape_array = model->GetArray(stb_op->inputs[1]); in Run() local191 if (!block_shape_array.buffer) { in Run()194 CHECK_EQ(block_shape_array.shape().dimensions_count(), 1); in Run()196 block_shape_array.Array::GetBuffer<ArrayDataType::kInt32>().data[0]; in Run()
1284 const auto& block_shape_array = model->GetArray(op->inputs[1]); in ProcessSpaceToBatchNDOperator() local1286 const auto& block_shape_array_shape = block_shape_array.shape(); in ProcessSpaceToBatchNDOperator()1293 if (!block_shape_array.buffer) { in ProcessSpaceToBatchNDOperator()1296 QCHECK(block_shape_array.data_type == ArrayDataType::kInt32); in ProcessSpaceToBatchNDOperator()1298 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessSpaceToBatchNDOperator()1333 const auto& block_shape_array = model->GetArray(op->inputs[1]); in ProcessBatchToSpaceNDOperator() local1335 const auto& block_shape_array_shape = block_shape_array.shape(); in ProcessBatchToSpaceNDOperator()1342 if (!block_shape_array.buffer) { in ProcessBatchToSpaceNDOperator()1345 QCHECK(block_shape_array.data_type == ArrayDataType::kInt32); in ProcessBatchToSpaceNDOperator()1347 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessBatchToSpaceNDOperator()