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::OkStatus(); 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::OkStatus(); 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()
188 const auto& block_shape_array = model->GetArray(stb_op->inputs[1]); in Run() local189 if (!block_shape_array.buffer) { in Run()192 CHECK_EQ(block_shape_array.shape().dimensions_count(), 1); in Run()194 block_shape_array.Array::GetBuffer<ArrayDataType::kInt32>().data[0]; in Run()
1296 const auto& block_shape_array = model->GetArray(op->inputs[1]); in ProcessSpaceToBatchNDOperator() local1298 const auto& block_shape_array_shape = block_shape_array.shape(); in ProcessSpaceToBatchNDOperator()1305 if (!block_shape_array.buffer) { in ProcessSpaceToBatchNDOperator()1308 QCHECK(block_shape_array.data_type == ArrayDataType::kInt32); in ProcessSpaceToBatchNDOperator()1310 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessSpaceToBatchNDOperator()1350 const auto& block_shape_array = model->GetArray(op->inputs[1]); in ProcessBatchToSpaceNDOperator() local1352 const auto& block_shape_array_shape = block_shape_array.shape(); in ProcessBatchToSpaceNDOperator()1359 if (!block_shape_array.buffer) { in ProcessBatchToSpaceNDOperator()1362 QCHECK(block_shape_array.data_type == ArrayDataType::kInt32); in ProcessBatchToSpaceNDOperator()1364 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessBatchToSpaceNDOperator()